AIML - Machine Learning Engineer, Foundation Models jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 1 week ago

AIML - Machine Learning Engineer, Foundation Models

positionSan Francisco, CA
timeFull-time
remoteOnsite
senioritySenior Level
money$172K/yr - $302K/yr
date5+ years exp

Apple is a company that pushes the boundaries of computing and intelligence, building products that bring smiles to people’s faces. They are seeking a Machine Learning Engineer to work with the Foundation Model Infrastructure team to optimize inference for cutting-edge model architectures and build production-grade solutions for real-time customer service.

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

Responsibilities

Work along side Foundation Model Research team to optimize inference for cutting edge model architectures
Work closely with product teams to build Production grade solutions to launch models serving millions of customers in real time
Build tools to understand bottlenecks in Inference for different hardwares and use cases
Mentor and guide engineers in the organization

Qualification

Machine LearningCloud ServicesGPU ProgrammingDeep Learning ArchitecturesModern Programming LanguagesML FrameworksMentoringProject Management

Required

5+ years of experience leading and driving complex, ambiguous projects
Have experience with high throughput services particularly at supercomputing scale
Proficient with running applications on Cloud (AWS / Azure or equivalent) using Kubernetes, Docker etc
Familiar with GPU programming concepts using CUDA
Familiar with one of the popular ML Frameworks like Pytorch, Tensorflow

Preferred

Proficient in building and maintaining systems written in modern languages (eg: Golang, python)
Familiar with fundamental Deep Learning architectures such as Transformers, Encoder/Decoder models
Familiarity with Nvidia TensorRT-LLM, vLLM, DeepSpeed, Nvidia Triton Server etc
Experience writing custom CUDA kernels using CUDA or OpenAI Triton

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