AIML - ML Infrastructure Engineer, ML Platform & Technology - ML Compute jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 2 months ago

AIML - ML Infrastructure Engineer, ML Platform & Technology - ML Compute

positionSeattle, Washington, United States
timeFull-time
remoteOnsite
senioritySenior Level, Lead/Staff
money$172K/yr - $302K/yr
date5+ years exp

Apple is where individual imaginations gather together, committing to the values that lead to great work. As a Senior/Staff Engineer on the Foundation Model Compute Infra team, you will design and scale the scheduling and orchestration systems that power Apple’s large-scale foundation model training and inference workloads across TPU clusters.

AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Design and scale the scheduling and orchestration systems that power Apple’s large-scale foundation model training and inference workloads across TPU clusters
Drive innovations in resource management, cluster efficiency, and platform reliability, enabling Apple’s next-generation AI models to train and serve at scale

Qualification

Machine LearningDistributed SystemsCloud ComputingPythonKubernetesTensorFlowCommunication SkillsTeam Collaboration

Required

Bachelors in Computer Science, engineering, or a related field
5+ years of hands-on experience in building scalable backend systems for training and evaluation of machine learning models
Proficient in relevant programming languages, like Python or Go
Strong expertise in distributed systems, reliability and scalability, containerization, and cloud platforms
Proficient in cloud computing infrastructure and tools: Kubernetes, Ray, PySpark
Ability to clearly and concisely communicate technical and architectural problems, while working with partners to iteratively find

Preferred

Advance degrees in Computer Science, engineering, or a related field
Proficient in working with and debugging accelerators, like: GPU, TPU, AWS Trainium
Proficient in ML training and deployment frameworks, like: JAX, Tensorflow, PyTorch, TensorRT, vLLM

Benefits

Comprehensive medical and dental coverage
Retirement benefits
A range of discounted products and free services
Reimbursement for certain educational expenses — including tuition
Discretionary bonuses or commission payments
Relocation

Company

Apple

twittertwittertwitter
Glassdoor4.2
company-logo
Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.
dateFounded in 1976
location
Cupertino, California, USA
people-size10001+ employees
web-link
https://www.apple.com

H1B Sponsorship

Apple has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (6998)
2024 (3766)
2023 (3939)
2022 (4822)
2021 (4060)
2020 (3656)

Funding

Current Stage
Public Company
Total Funding
$5.67B
Key Investors
Berkshire HathawayMicrosoftSequoia Capital
2025-05-05Post Ipo Debt· $4.5B
2025-01-16Post Ipo Debt· $0.31M
2021-04-30Post Ipo Equity

Leadership Team

leader-logo
Tim Cook
CEO
leader-logo
Craig Federighi
SVP, Software Engineering

Recent News

Venrock
Venrock Portfolio
2025-12-01
IndiaTimes
Apple TV+ pulls Jessica Chastain’s "The Savant" just days before premiere
2025-09-25
Mac Daily News
Jessica Chastain: I respect Apple’s decision to pause the release of ‘The Savant’ or something
2025-09-25
Company data provided by crunchbase