Apple · 5 months ago
Senior AI Infra Performance Engineer
Santa Clara, California, United States
Full-time
Onsite
Senior Level
$181K/yr - $318K/yr
7+ years expApple is a leading technology company known for its innovative products and services. The Senior AI Infra Performance Engineer will work in the Machine Learning Platform Technologies organization, focusing on building infrastructure for large-scale ML training and inference workloads, while addressing performance challenges and optimizing resource utilization.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Characterize ML workload behavior through profiling, benchmarks and metrics
Dive into unfamiliar codebases to prototype changes, evaluate tradeoffs, and build production-ready solutions
Design systems for efficient recovery from failures and preemptions
Create tools to identify and alert bottlenecks across applications and frameworks
Use workload-driven insights to influence next-generation hardware selection and procurement decisions
Collaborate closely with ML researchers and infrastructure engineers to address inefficiencies
Drive impact through hands-on contribution and mentorship
Qualification
Required
Experience with large-scale distributed systems for AI/ML workloads running on GPUs or TPUs
Strong software engineering skills with experience developing and optimizing training frameworks (e.g. PyTorch, JAX) using C/C++ or Python
Experience working on cross-functional projects with ML research and infrastructure teams
Familiarity with model architectures and various training techniques
Bachelor's degree in Computer Science or equivalent experience, with 7+ years of industry experience
Preferred
Have a track record of delivering transformative performance improvements on large scale infrastructure
Ability to analyze ambiguous, distributed systems problems and articulate both high-level strategic metrics and underlying technical complexity
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
Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.
10001+ employees
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 CompanyTotal Funding
$5.67BKey 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
Tim Cook
CEO
Craig Federighi
SVP, Software Engineering
Recent News
Venrock
2025-12-01
2025-09-25
Mac Daily News
2025-09-25
Company data provided by crunchbase