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!

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., 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.

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).

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.