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 features.
Domain
Consumer technology, Generative AI, Foundation Models
Deliverable
production ML models
Required skills
dataset construction, data annotation, data enrichment, data curation, synthetic data generation, ML fundamentals, applied research, image/video data handling, diffusion model training, auto-regressive model training, data ablation experiments
Preferred skills
diffusion image generation model improvement, industrial research output, shipped complex projects, top-tier venue publications
Technologies
PyTorch, JAX, Hadoop, Ray