Apple · 4 months ago
AIML - Staff Machine Learning Engineer - ML Efficiency, ML Platform & Technology
Santa Clara, California, United States
Full-time
Onsite
Senior Level, Lead/Staff
$172K/yr - $302K/yr
7+ years expApple is seeking a highly motivated and experienced Staff Machine Learning Engineer to join their Machine Learning Platform Technology & Infra team. The role focuses on optimizing end-to-end system performance of distributed machine learning workloads and involves collaboration with ML researchers and key partners across the company.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Engage with ML researchers to optimize end-to-end performance of large scale distributed ML workloads
Analyze workload metrics to identify sources of inefficiencies and work with users to understand and optimize ML workloads
Conduct workload analysis based on benchmarking key workloads on deployed systems
Improve large scale training resiliency by optimizing applications and frameworks for improved recovery from failures and preemptions
Influence architecture, design, development, and operations of next generation ML accelerator systems based on workload insights
Qualification
Required
Experience working with large scale parallel and distributed accelerator-based systems
Experience optimizing performance and AI workloads at scale
Experience developing code in one or more of training frameworks (such as PyTorch, TensorFlow or JAX)
Strong communicator with ability to analyze complex and ambiguous problems
Programming and software design skills (proficiency in C/C++ and/or Python)
Experience working in a high-level collaborative environment and promoting a teamwork mentality
Bachelor's degree in Computer Science and 7+ years of work experience
Preferred
Deep understanding of computer systems and the interactions between HW and SW
Experience in performance analysis and optimization experience in Cloud accelerators
Advanced degree in CS
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