AIML - Staff ML Infrastructure Engineer, Machine Learning Compute

Apple is seeking a Staff ML Infrastructure Engineer to join their Machine Learning Compute team. This role combines deep technical expertise in distributed systems and ML infrastructure with leadership responsibilities. You'll be at the forefront of building and optimizing large-scale ML compute platforms that power Apple's AI initiatives.

As a staff engineer, you'll lead the development of infrastructure for cloud-based workloads, working with technologies like Apache Spark, Ray, and distributed training systems. You'll be responsible for optimizing platform efficiency through advanced resource management, implementing new features from core frameworks, and enhancing system observability.

The role requires both technical excellence and leadership skills. You'll drive architectural decisions that improve system performance and scalability while mentoring other engineers. You'll work with cutting-edge cloud-native technologies and have the opportunity to shape the future of Apple's ML infrastructure.

The ideal candidate brings 4+ years of hands-on experience with large-scale data and ML infrastructure, strong programming skills in Python or Go, and deep expertise in distributed systems and cloud platforms. You should be comfortable with technologies like Kubernetes, Apache Spark, and Ray, and have excellent communication skills.

At Apple, you'll be part of a team that values innovation and collaboration. You'll receive comprehensive benefits including medical coverage, retirement benefits, stock programs, and education reimbursement. The base salary range is $181,100 to $318,400, with additional opportunities for bonuses and equity.

This is an opportunity to make a significant impact on Apple's ML infrastructure while working with some of the most advanced technologies in the field. You'll be part of a company that values diversity, inclusion, and the ability to make products that change lives for the better.