Distinguished Engineer, AI Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 7 hours ago

Distinguished Engineer, AI Infrastructure

Apple is where individual imaginations gather together, committing to the values that lead to great work. As part of the Machine Learning Platform Technologies - ML Compute organization, you’ll connect the world’s best researchers with the world’s best AI infrastructure to take on the most challenging problems in machine learning and build the rock-solid foundation for some of Apple’s most innovative products.

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

Responsibilities

Set the technical vision and define the roadmap for frontier AI infrastructure spanning a multi-cloud heterogeneous compute fleet
Design and build distributed systems for job scheduling, cluster management, and cost governance
Partner closely with ML researchers and engineers to scale and optimize end-to-end performance of AI training and inference workloads
Establish connections with cross-functional partners, collaborating with teams across cloud infrastructure, business operations, finance, and engineering

Qualification

Large-scale distributed systemsBackend programming languagesCloud-native architecturesML accelerators experienceML systems performance profilingDeep learning frameworksMentoring senior engineersTechnical leadership

Required

Bachelor's degree in Computer Science, relevant technical field, or equivalent practical experience
15 years of experience designing and building very large-scale distributed systems
Proficiency in at least 1 backend language (e.g., Python, C++, Go, Rust)
Proficiency in cloud-native architectures and compute orchestration platforms (e.g., Kubernetes)
Hands-on experience working with ML accelerators such as GPUs and TPUs
Track record of mentoring senior engineers and setting long-term architectural vision across orgs

Preferred

Master's degree or PhD in Computer Science or related technical fields
Experience in running distributed training and/or inference workloads in production
Expertise in ML systems performance profiling, debugging, and optimization
Expertise in deep learning frameworks (e.g., PyTorch, JAX) and their underlying architectures

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 is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.

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
Company data provided by crunchbase