AIML - Senior Machine Learning Infrastructure Engineer -ML Compute, ML Platform & Technology

Apple is seeking a Senior Machine Learning Infrastructure Engineer to join their ML Compute Team, focusing on building and maintaining compute infrastructure for ML workloads. This role combines cutting-edge machine learning infrastructure with large-scale distributed systems engineering.

The ML Compute Team is responsible for building and maintaining compute infrastructure to run ML workloads at Apple, including training, inferencing, and fine-tuning. As a Senior Engineer, you'll work on solving unique, large-scale, highly complex technical problems to make ML workloads more stable, reliable, efficient, and cost-effective. You'll collaborate with teams across Apple, impacting hundreds of millions of users through critical features essential to current and future Apple products.

Your work will involve:

• Architecting and implementing scalable backend systems for ML workloads
• Building robust infrastructure using Go and Python
• Working with cutting-edge ML technologies and frameworks
• Collaborating with teams across Apple on various ML initiatives
• Mentoring team members and conducting code reviews
• Driving the design and delivery of critical features

The ideal candidate will have strong expertise in distributed systems, cloud platforms, and ML infrastructure tools like Kubernetes, Ray, and Beam. You'll need excellent communication skills to work effectively with partners across Apple and help shape the future of ML infrastructure at scale.

Benefits include competitive base pay ($143,100-$264,200), equity through stock programs, comprehensive medical coverage, retirement benefits, education reimbursement, and potential bonuses. Join Apple in building the next generation of ML infrastructure that powers innovations used by millions worldwide.