← All roles

AIML - Machine Learning Engineer, Foundation Models

Apple· Machine Learning and AI
Apply on company site
Cupertino, CA$184.7k - $324.8kFull-timePosted 6mo ago

About the role

Apple is building foundation models optimized for on-device, private, and deeply integrated experiences. The role focuses on designing scalable ML infrastructure for foundation models across text, image, speech, and multi-modal domains, productionizing state-of-the-art AI algorithms, and ensuring privacy and security. May lead a small team of engineers.

Responsibilities

  • Design and implement scalable, reliable and high-performance machine learning infrastructure for foundation models across text, image, speech, and multi-modal domains
  • Collaborate with other teams to productionize state-of-the-art AI algorithms
  • Optimize models for performance, efficiency, and on-device intelligence
  • Implement machine learning systems with stringent privacy and security requirements
  • May also be required to manage a small team of engineers.

Qualifications

  • MS or PhD in Computer Science, Machine Learning, or related technical field
  • Expert-level programming skills in Python
  • Proficiency in machine learning frameworks such as Jax, PyTorch, TensorFlow
  • Strong background in: Distributed training, Model optimization, and Machine learning infrastructure
  • Experience with large-scale model training and deployment
  • Familiarity with: Kubernetes, Docker, Cloud platforms (AWS, GCP, Azure)
  • Distributed computing frameworks
  • Experience with foundation models and large language models
  • Background in multi-modal AI systems
  • Demonstrated ability to transform research prototypes into production systems
  • Published research or significant contributions to open-source ML projects
  • Understanding of on-device machine learning techniques

Skills

Required:PythonJaxPyTorchDistributed trainingModel optimizationOn-device ML
Nice to have:TensorFlowKubernetesDocker

Benefits

  • Comprehensive medical and dental coverage
  • retirement benefits
  • discounted products and free services
  • reimbursement for certain educational expenses (tuition)
  • discretionary bonuses or commission payments and relocation may be available
  • stock programs including discretionary RSU awards and Employee Stock Purchase Plan
Last updated Oct 9, 2026. Interested? Apply on the company site.

About Apple

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

AirPods Max 2Apple ArcadeApple CardApple Fitness+Apple MusicApple TVApple TV+Apple Watch Series 11
Company profileWebsite

More open roles at Apple