On-device ML Infrastructure Engineer (ML Insights and Forecasting)

The On-Device Machine Learning team at Apple is seeking an experienced ML Infrastructure Engineer focused on ML Insights and Forecasting. This role sits at the heart of Apple's on-device machine learning initiatives, working across research, software engineering, hardware engineering, and products.

The team is responsible for building critical infrastructure that enables the Research to Production lifecycle of innovative machine learning models powering user experiences across Apple's platforms. This includes onboarding new ML architectures to embedded devices, developing optimization toolkits, creating ML compilers and runtimes, and building comprehensive benchmarking and debugging toolchains.

As an ML Infrastructure Engineer, you'll explore cutting-edge trends in ML architectures and ensure they run effectively on Apple's on-device stack. You'll build infrastructure for regular testing coverage and work closely with inference stack teams to optimize performance and enable new capabilities. The role offers exciting opportunities to dive into on-device ML research areas like model visualization, efficient inference algorithms, model compression, and ML compilers/runtime systems.

The position requires strong technical skills in Python programming and ML frameworks, along with solid fundamentals in machine learning concepts and quality practices. You'll collaborate extensively with ML and hardware teams across Apple to identify and address gaps in the ML inference stack.

This is an excellent opportunity to work at the forefront of on-device machine learning, contributing to infrastructure that underpins critical ML workflows across Apple's key products including Camera, Siri, Health, Vision, and more. You'll be part of Apple Intelligence, helping to advance the state-of-the-art in efficient, on-device machine learning capabilities.

The role offers competitive compensation including base pay between $139,500 and $258,100, plus additional benefits like stock programs, comprehensive healthcare, education reimbursement, and potential bonuses. Join Apple's world-class team working on next-generation machine learning technologies that directly impact hundreds of millions of users.