AIML - ML Infrastructure Engineer, ML Platform & Technology - ML Compute

Apple is seeking a Staff ML Infrastructure Engineer to join their ML Platform & Technology team, focusing on ML Compute initiatives. This role sits at the intersection of machine learning and distributed systems, where you'll be instrumental in building and scaling the infrastructure that powers Apple's most complex ML models.

As a staff engineer, you'll lead critical initiatives in large-scale ML training, working with cutting-edge technologies and frameworks. You'll be responsible for architecting and implementing robust MLOps platforms, optimizing distributed training systems, and ensuring seamless operation of ML workloads at scale. The role requires expertise in both ML infrastructure and distributed systems, with hands-on experience in technologies like Kubernetes, Ray, and cloud computing platforms.

The ideal candidate brings strong technical depth in ML systems and distributed computing, coupled with the ability to lead complex technical projects and mentor other engineers. You'll work in a collaborative environment, partnering with cross-functional teams to solve challenging problems in ML infrastructure and training optimization.

This position offers competitive compensation, including base pay between $147,400 and $220,900, plus additional benefits like stock awards, comprehensive healthcare, and education reimbursement. You'll be part of Apple's innovative culture where individual imaginations come together to create products that change lives for the better.

The role is based in the San Francisco Bay Area, where you'll join a team committed to technical excellence and innovation. If you're passionate about building scalable ML infrastructure and want to impact how machine learning is deployed at one of the world's most influential technology companies, this role offers an exceptional opportunity to make your mark.