AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure

A leading technology company that creates innovative products, services, and customer experiences.
$207,800 - $378,700
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
10+ years of experience
AI

Description For AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure

At Apple, we're seeking a Machine Learning Engineer to join our Information Intelligence Infrastructure team. This role is at the forefront of building groundbreaking technology for search, natural language processing, artificial intelligence, and machine learning. Our infrastructure serves as the backbone of Apple Intelligence, powering the largest Apple foundation models and a wide range of services including Apple Search, Apple Music, AppleTV, AppStore, iMessages, Photos & Camera, Spotlight, Safari, and Siri.

You'll work in one of the most exciting high-performance computing environments, handling petabytes of data and millions of queries per second. Your role involves optimizing billions of parameter language, vision, and speech models using state-of-the-art technologies at Apple's scale. The team processes billions of requests daily across our search and foundation model platform, taking full end-to-end ownership of services from conception through maintenance.

As a Machine Learning Engineer, you'll collaborate with the Foundation Model Research team to optimize inference for cutting-edge model architectures and work closely with product teams to develop production-grade solutions. You'll be responsible for building and maintaining systems using modern languages like Golang and Python, working with fundamental Deep Learning architectures, and potentially developing custom CUDA kernels.

The position offers a competitive base pay range of $207,800 to $378,700, along with comprehensive benefits including medical and dental coverage, retirement benefits, stock options, and education reimbursement. Join us in creating intelligent solutions that delight millions of Apple users every day while working with cutting-edge technology and brilliant colleagues in a collaborative environment.

Last updated 20 hours ago

Responsibilities For AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure

  • Design, build and maintain infrastructure to support features that empower billions of Apple users
  • Process billions of requests every day across search and foundation model platform
  • Take full end-to-end ownership of services through conception, design, implementation, deployment, and maintenance
  • Work on complex large scale systems with trillions of records and petabytes of data
  • Work alongside Foundation Model Research team to optimize inference for cutting edge model architectures
  • Work closely with product teams to build production grade solutions for millions of customers in real time

Requirements For AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure

Python
Go
Kubernetes
  • Strong background in computer science: algorithms, data structures and system design
  • 10+ year experience on large scale distributed system design, operation and optimization
  • Familiar with one of the popular ML Frameworks like Pytorch, Tensorflow
  • Excellent interpersonal skills able to work independently as well as cross-functionally

Benefits For AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure

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

Interested in this job?

Jobs Related To Apple AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure

Staff Machine Learning Engineer, Apple ML Data Platform

Staff Machine Learning Engineer position at Apple focusing on ML infrastructure and generative AI, building scalable data pipelines and advanced ML workflows.

Machine Learning Engineer- Advanced Analytics, Worldwide Business Process Re-Engineering

Senior Machine Learning Engineer role at Apple focusing on implementing Generative AI solutions for supply chain optimization, offering competitive compensation and comprehensive benefits.

CPU Research Scientist - Platform Architecture

Research Scientist role at Apple focusing on CPU architecture research and development for next-generation iPhone, iPad, and Mac processors.

Senior Multimodal Generative Modeling Research Engineer - SIML, ISE

Senior ML Engineering role at Apple focusing on multimodal generative AI models, requiring 5+ years of experience and expertise in distributed training and model optimization.

AIML - Staff Machine Learning Engineer, Retrieval and Ranking Quality (SII)

Staff Machine Learning Engineer position at Apple focused on developing retrieval and ranking systems using advanced ML/NLP techniques.