Machine Learning Engineer — Generative Models, Productivity Apps
At a glance
AI generatedTL;DR
As a Machine Learning Engineer at Apple’s Productivity Apps team, you will join a world-class engineering group to develop cutting-edge generative models and applications that enhance productivity tools like Notes and iWork. Your day-to-day responsibilities include designing, training, and evaluating advanced generative models such as diffusion and transformer architectures for creative workflows, collaborating with cross-functional teams to deliver end-to-end features from research prototypes to production, and staying abreast of the latest research trends to implement novel ideas in software development. This role requires expertise in PyTorch or JAX, a deep understanding of generative modeling techniques like GANs and VAEs, and experience deploying models in constrained environments. Ideal candidates have a background in Computer Science with relevant industry experience and publications in top-tier conferences, driving innovation in the productivity space at scale.
Skills
What you'll do
- Design, train, and evaluate advanced generative models for productivity tool applications.
- Develop innovative model architectures to generate professional-level visual outputs.
- Partner with cross-functional teams to deliver end-to-end features from prototype to production.
- Stay updated with the latest research trends in generative modeling and implement new ideas.
- Contribute to top-tier publications in machine learning and computer vision conferences.
What we're looking for
- MS + 2 years industry experience or PhD in relevant field with generative model publications
- Deep hands-on experience training diffusion models, GANs, VAEs, and autoregressive image models
- Strong programming skills in PyTorch or JAX for machine learning applications
- Ability to design and evaluate novel model architectures for professional creative workflows
- Experience partnering cross-functionally to deliver end-to-end features from prototype to production
Employer
About Apple Inc
Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software
Apple Inc currently has 255 open roles on FindRole.
Listed pay typically runs $171,600–$272,100 across 182 roles with salary data.
Most-posted roles
- Software Development Engineer 10
- Apple Business Systems Engineer Manager 8
- iPad Touch Electrical Engineer 3
- Machine Learning Engineer, Apple Store Online 3
- Manager, Machine Learning, Apple Store Online 3