Apple · 1 day ago
AIML - Staff Software Engineer, ML Compute - ML Platform & Technologies
Seattle, WA
Full-time
Onsite
Lead/Staff
$172K/yr - $302K/yr
10+ years expApple is a leading technology company dedicated to innovation and creativity. They are seeking a Staff Software Engineer to design, build, and maintain the compute infrastructure for machine learning and AI applications, collaborating with teams to deliver high-quality products and mentoring junior engineers.
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. In this role, you are building the platform upon which other teams will develop data pipelines and machine learning applications
Lead a group of engineers to deliver high-quality products/services. Be able to stay on top of innovative technologies and apply them in the job
During the process, support junior engineers by providing advice, mentoring, and educational opportunities
Qualification
Required
Bachelors in Computer Science
10+ years of industry related experience, working in collaborate 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