Apple · 3 hours ago
AIML - Staff Software Engineer, ML Compute - ML Platform u0026 Technologies
Seattle, WA
Full-time
Onsite
Lead/Staff
$172K/yr - $302K/yr
10+ years expApple is where individual imaginations gather together, committing to the values that lead to great work. As part of the ML Compute team, you will be at the heart of designing, building, and maintaining the compute infrastructure that enables model training, tuning, and other ML tasks for machine learning, artificial intelligence, and computer vision applications.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Own the architecture, design, development, and operations of large-scale systems designed for machine learning
Develop custom scheduling, resource management solutions, and fleet management for our ML model training compute infrastructure
Collaborate with multi-functional teams, integrate with Kubernetes in on-premises and cloud provider clusters, and enable seamless integration with NVIDIA GPUs and other ML accelerators
Partner with data scientists and machine learning engineers across different Apple organizations to define high-impact product features and deliver them with quality
Lead a group of engineers to deliver high-quality products/services
Support junior engineers by providing advice, mentoring, and educational opportunities
Qualification
Required
Bachelors in Computer Science
10+ years of industry related experience, working in collaborative environments
Experience with public cloud infrastructure like: Kubernetes, Amazon EC2 and EKS, Google Cloud Platform
Strong software development skills, with proficiency in relevant languages (ex. Golang, Python)
Strong problem solving skills and ability to write performant and high-quality code
Proficient understanding of the software development process, including unit testing and release management
Excellent interpersonal skills; able to work independently as well as in a team; can take feedback and iterate on a solution in a collaborative setting
Preferred
Strong understanding of batch scheduling systems and high-performance computing environments
Experience using system monitoring tools, automated testing frameworks and CI/CD pipelines
Experience with GPUs and/or other ML accelerators in the context of Machine Learning
A passion for making simple, robust, and scalable platforms used by other engineering teams
Flexibility/adaptability for working in a dynamic environment with different frameworks and requirements
Masters or PhD in Computer Science
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