Apple · 3 months ago
AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - ML Compute
San Francisco Bay Area
Full-time
Onsite
Senior Level, Lead/Staff
$181K/yr - $318K/yr
6+ years expApple is where individual imaginations gather together, committing to the values that lead to great work. As a staff engineer on the ML Compute team, your work will include leading the development of infrastructure for large-scale workloads on the Cloud, optimizing platform efficiency, and mentoring engineers while driving the architectural evolution of the platform.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Lead the development of the infrastructure to run large-scale workloads on the Cloud, such as Apache Spark, Ray, and distributed training
Optimize platform efficiency and throughput by improving resource management capabilities with schedulers like Apache YuniKorn and Kueue
Integrate new features from core distributed computing and ML frameworks into the platform, offering them to production users and providing support
Enhance the platform's scalability, performance, and observability through improved monitoring and logging
Drive the architectural evolution of the platform by adopting modern, cloud-native technologies to improve system performance, efficiency, and scalability
Reduce dev-ops efforts by automating and streamlining operational processes
Mentor engineers in areas of your expertise, fostering skill growth and knowledge sharing
Qualification
Required
Bachelors in Computer Science, engineering, or a related field
6+ 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 solutions
Preferred
Advance degrees in Computer Science, engineering, or a related field
Hands-on experience with cloud-native resource management and scheduling tools like Apache YuniKorn
Experience with advanced architecture for distributed data processing and ML workloads
Proficient in working with and debugging accelerators, like: GPU, TPU, AWS Trainium
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
Late StageTotal Funding
$5.67BKey Investors
Berkshire HathawayMicrosoftSequoia Capital
2026-01-10Pre Seed· $1M
2025-05-05Post Ipo Debt· $4.5B
2025-01-16Post Ipo Debt· $0.31M
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