Machine Learning Engineer, Data for Foundation Models
Core
Building high-quality datasets for foundation model training to improve diffusion and auto-regressive models for Apple products like Genmoji and Image Playground.
Role type
Senior IC machine learning research engineer (data for foundation models)
Builds
Web-scale datasets, distributed training pipelines, application-specific adapters, and on-device optimizations for generative AI.
Domain
Generative AI, Foundation Models, Image/Video Understanding
Deliverable
production ML models
Required skills
dataset construction, data annotation, data enrichment, synthetic data generation, diffusion model training, auto-regressive model training, data ablation experiments, PyTorch, JAX, Hadoop, Ray
Preferred skills
improving diffusion image generation model quality via training data, strong industrial research output, shipped complex projects, top-tier venue publications