Taro Logo

AIML - Machine Learning Engineer, Foundation Models

Apple is revolutionizing artificial intelligence by developing sophisticated foundation models that power intelligent features across their product ecosystem.
$171,600 - $302,200
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI

Job Description

Apple is seeking exceptional Machine Learning Engineers to join their Foundation Model Team, focusing on developing sophisticated AI models that power intelligent features across Apple's product ecosystem. This role represents an opportunity to work at the intersection of cutting-edge AI research and large-scale production systems.

The Foundation Model Team is responsible for building fundamental AI capabilities in text, images, speech, and multi-modal applications. As a Machine Learning Engineer, you'll be instrumental in designing and implementing scalable infrastructure for training and deploying these advanced models while ensuring they meet Apple's high standards for performance and privacy.

The role combines deep technical expertise with practical engineering skills, requiring proficiency in Python, distributed systems, and modern ML frameworks. You'll work on challenging problems in model optimization, distributed training, and efficient deployment, directly impacting millions of users through Apple's products.

Key aspects of the role include:

  • Building and optimizing large-scale foundation models
  • Developing efficient training and inference systems
  • Collaborating across teams to integrate AI capabilities into products
  • Ensuring robust privacy and security measures
  • Potential team leadership opportunities

The position offers competitive compensation ($171,600 - $302,200) and comprehensive benefits including medical coverage, stock options, and education reimbursement. Multiple locations are available including Cupertino, New York City, and Seattle, offering flexibility in choosing your work location.

This is an excellent opportunity for those passionate about advancing the field of AI while working on products that impact millions of users daily. The role combines the excitement of research-driven innovation with the satisfaction of seeing your work deployed in real-world applications at massive scale.

Last updated 2 days ago

Responsibilities For AIML - Machine Learning Engineer, Foundation Models

  • 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

Requirements For AIML - Machine Learning Engineer, Foundation Models

Python
Kubernetes
  • 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

Benefits For AIML - Machine Learning Engineer, Foundation Models

Medical Insurance
Dental Insurance
Vision Insurance
401k
Education Budget
Equity
Relocation Benefits
  • Comprehensive medical and dental coverage
  • Retirement benefits
  • Discounted products and free services
  • Education reimbursement
  • Stock options and employee stock purchase plan
  • Discretionary bonuses
  • Relocation assistance

Related Jobs

AIML - Sr. Software Engineer - AIML Observability

Senior Software Engineer role at Apple focusing on AI/ML observability, building cloud-native solutions for monitoring and visualization of AI infrastructure at scale.

Cellular Machine Learning Engineer - Embedded Software

Senior Machine Learning Engineer role at Apple focusing on developing AI/ML solutions for cellular technologies in iPhone, iPad, and Watch products.

AIML - Sr. Software Engineer - AIML Observability

Senior Software Engineer role focused on building AI-powered observability solutions at Apple, working on cloud-native systems and AIML infrastructure.

Computer Vision/ Machine Learning Software Engineer - Apple Vision Pro

Senior Computer Vision/ML Engineer role at Apple working on Vision Pro's ARKit algorithms team, developing real-time visual perception and localization systems.

AIML - Machine Learning Engineer, Foundation Models

Senior Machine Learning Engineer position at Apple focusing on foundation models and AI development