Edge Ai Machine Learning Jobs in Rohnert Park, CA

iPhone is the most popular camera in the world. The flawless integration of software and hardware has led to features like Memories and Portrait Mode which deliver experiences that are magical. The Camera & Photos team focuses on user-experience by using computer vision and image processing through machine learning. Our team works hard on products that ship to millions of people. We are looking for people who want to do the same.
The Computational Imaging team develops the camera pipelines and innovative algorithms for Apple’s mobile devices, including the iPhone and iPad. Combining innovative algorithms with optimized implementations, our team delivers the quality and features which help to re-define mobile photography. In this role, you will work hands-on with diffusion models, transformers and next-gen generative techniques, while collaborating closely with product, design and research teams to bring your ideas to life. If you are passionate about inventing and developing new imaging algorithms to improve the photography experience, we would like to hear from you.
Description
Design novel generative model architectures and training strategies for vision and computational photography applications.
Fine-tune and adapt pre-trained models for specific use cases and domains.
Build scalable and robust data pipelines for large-scale image and video datasets.
Develop comprehensive evaluation frameworks including implementing quantitative metrics and qualitative analysis methodologies, conducting ablation studies and hyper-parameter optimization, for model performance assessment.
Conduct deep technical analysis and identify research directions.
Participate in code reviews, technical discussions and knowledge sharing sessions.
Communicate complex technical concepts through clear documentation and presentations.
Stay current with latest research in generative AI, computer vision and computational photography. Translate findings into practical applications.
Preferred Qualifications
Minimum PhD + 2 years or MS + 5 years industry experience.
Expertise in image processing, computer vision and deep learning.
Strong experience with deep learning frameworks (e.g. PyTorch) and modern ML toolchains.
Proven experience with training and developing generative models.
Hands-on with multi-GPU/multi-node/distributed training and large-scale experimentation.
Curiosity-driven with a track-record of identifying and solving complex technical problems.
Excellent communication skills and an ability to bridge gap between technical depth and product impact.
Collaborative mindset with openness to feedback and continuous learning.
Publications at major conferences (CVPR, ICCV, NeurIPS, ICML) or open-source contributions would be a plus.
Minimum Qualifications
PhD or MS + 3 years industry experience.
Solid foundation in image processing, computer vision and deep learning fundamentals.
Academic or project experience with image/video generation models.
Pay & Benefits
At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $150,400 and $277,600, and your base pay will depend on your skills, qualifications, experience, and location.
Apple employees also have the opportunity to become an Apple shareholder through participation in Apple's discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple's Employee Stock Purchase Plan. You'll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses - including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.