Machine Learning Engineer - Generative Models, Productivity Apps
At Apple, new ideas have a way of becoming phenomenal products, services, and customer experiences very quickly! The Productivity Apps team, the team behind apps like Notes, Freeform, and iWork needs your help shaping the next generation of productivity tools by working on pioneering technologies to surprise and delight our users. As a Machine Learning Engineer, you will be working alongside our world-class creatives, designers, and engineers to help innovate in the productivity space in ways that only Apple can. This is a highly visible, highly impactful opportunity!
Description
Join our research-oriented engineering team, and you'll build state-of-the-art generative models and applications, partner with cross-functional teams, and deliver end-to-end features to power the next-generation creative tools. The ideal candidate should have deep experience in generative modeling, care about long-term sustainable software development, and can drive features from concept all the way to delivery. This position requires a self-motivated individual with excellent interpersonal skills to effectively collaborate with all levels of the organization.
Responsibilities:
Want more jobs like this?
Get Software Engineering jobs in Cupertino, CA delivered to your inbox every week.
Design, train, and evaluate generative models (diffusion, transformers) for creative applications in productivity tools
Develop novel model architectures and systems that generate structured visual and graphic design outputs suitable for professional creative workflows
Partner with cross-functional teams (product, design, on-device ML) to deliver end-to-end features from research prototype to production
Stay current with the research landscape, read and implement ideas from papers, and contribute publications at top venues
Preferred Qualifications
Experience with compositional or layout-conditioned image generation, or image decomposition/editing
Experience deploying models to memory or compute constrained environments. Experience with large-scale model training and parallelization.
Track record of publications in top-tier machine learning or computer vision conferences (e.g., CVPR, ICCV, ECCV, NeurIPS/ICML, SIGGRAPH, CHI).
Minimum Qualifications
MS + 2 years of industry experience or PhD in Computer Science, Computer Vision, Machine Learning, or related field with publications in generative models
Hands-on experience training generative models such as diffusion models, GANs, VAEs, or autoregressive image models. Strong programming skills in PyTorch or JAX.
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 $147,400 and $272,100, 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.
Perks and Benefits
Health and Wellness
Parental Benefits
Work Flexibility
Office Life and Perks
Vacation and Time Off
Financial and Retirement
Professional Development
Diversity and Inclusion
Company Videos
Hear directly from employees about what it is like to work at Apple.