Machine Learning Engineer — Generative Models, Productivity Apps

Apple • Cupertino, CA

2d

About The Position

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.

Nice To Haves

• 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

• Build state-of-the-art generative models and applications.
• Partner with cross-functional teams.
• Deliver end-to-end features to power the next-generation creative tools.
• Drive features from concept all the way to delivery.

What This Job Offers

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