Machine Learning Engineer — Foundation Models
Core
Building foundation models with fundamental general capabilities (text, images, speech, video) and applying them to Apple products.
Role type
Senior IC machine learning researcher (foundation models)
Builds
Pre-trained foundation models, model architectures, and training infrastructure
Domain
Artificial Intelligence / Large Language Models / Deep Learning
Deliverable
production ML models
Required skills
Machine learning research, deep learning toolkits (JAX, TensorFlow, PyTorch), linear algebra, statistics, model architecture design, scientific scaling of LLMs, training optimization
Preferred skills
Research problem formulation, experiment design, publication record in top-tier conferences (NeurIPS, ICML, etc.), competitive programming
Technologies
JAX, TensorFlow, PyTorch
Responsibilities
Design efficient model architectures tailored to product requirements; propose and execute research plans to advance machine learning understanding; optimize workflows and infrastructure for training efficiency and scale; collaborate with broader teams across Apple
Seniority
Senior, hands-on IC
