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

As a Staff ML Infrastructure Engineer on the ML Compute team at Apple, you'll drive large-scale training initiatives, operationalize ML workloads on Kubernetes, enhance distributed cloud training techniques for foundation models, and design end-to-end lifecycles for distributed ML systems. You'll develop tools to optimize ML systems, architect a robust MLOps platform, and collaborate with cross-functional engineers to solve complex ML training challenges. Your role involves researching and implementing new technologies, leading technical projects, mentoring engineers, and fostering a collaborative, innovative team environment. This position offers the opportunity to work on cutting-edge ML infrastructure at one of the world's most innovative companies, with a competitive compensation package and comprehensive benefits.