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.
$181,100 - $318,400
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI

Job Description

Apple is revolutionizing artificial intelligence by developing sophisticated foundation models that power intelligent features across our product ecosystem. The Foundation Model Team is responsible for building infrastructure, datasets, and models with fundamental general capabilities such as understanding and generation of text, images, speech, videos, and other modalities.

As a Machine Learning Engineer, you will be instrumental in designing, developing, and deploying foundation models that drive intelligent features across Apple's product portfolio. You'll bridge the gap between research and production, transforming advanced AI concepts into robust, performant systems that enhance user experiences.

The role involves working with cutting-edge AI technologies, collaborating across teams, and implementing scalable solutions while maintaining Apple's high standards for privacy and security. You'll be part of a team pushing the boundaries of deep learning in terms of scaling, efficiency, and flexibility, with the opportunity to impact millions of users through Apple products.

This position offers competitive compensation including base pay between $181,100 and $318,400, plus equity opportunities through stock programs, comprehensive benefits, and potential bonuses. You'll work in a collaborative environment with some of the best minds in AI, contributing to projects that define the future of intelligent computing.

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
Equity
Education Budget
  • Comprehensive medical and dental coverage
  • Retirement benefits
  • Employee stock programs
  • Education reimbursement
  • Discretionary bonuses
  • Relocation assistance
  • Employee discount on Apple products

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 role at Apple focusing on foundation models, offering $171K-$302K, working on cutting-edge AI systems that power Apple's product ecosystem.