Machine Learning Engineer — Generative Models, Productivity Apps

Apple • Cupertino, CA

16h

About The Position

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.

Requirements

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

Responsibilities

• 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

What This Job Offers

• Job Type: Full-time
• Career Level: Senior