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

Who you are

You have a strong background in machine learning and infrastructure development, with experience in building and optimizing machine learning models for embedded devices. You are proficient in Python and have hands-on experience with machine learning frameworks such as TensorFlow. Your understanding of Kubernetes and Docker allows you to manage containerized applications effectively, ensuring smooth deployment and scalability. You are familiar with the latest trends in machine learning architectures and have a keen interest in exploring new technologies that can enhance on-device capabilities. You thrive in collaborative environments, working closely with cross-functional teams including software and hardware engineers to deliver innovative solutions. You are detail-oriented and possess strong analytical skills, enabling you to benchmark, analyze, and debug complex systems efficiently.

Desirable

Experience with additional machine learning tools and libraries is a plus, as is familiarity with optimization techniques for model performance on constrained devices. A background in research or a related field can further enhance your candidacy.

What you'll do

In this role, you will be responsible for enabling the Research to Production lifecycle of machine learning models that enhance user experiences across Apple’s hardware and software platforms. You will work on onboarding the latest machine learning architectures to embedded devices, developing optimization toolkits to tailor these models for specific targets. Your work will involve building machine learning compilers and runtimes to ensure efficient execution of models, as well as creating a robust benchmarking and debugging toolchain to facilitate continuous improvement of model iterations. You will explore new trends in machine learning architectures and integrate them into our on-device stack, contributing to the development of an end-to-end developer experience for ML development. Collaboration with research, software engineering, and hardware engineering teams will be key to your success, as you will help shape the future of on-device machine learning at Apple.

What we offer

Apple provides a dynamic work environment where innovation thrives. You will have the opportunity to work on cutting-edge technologies that impact millions of users worldwide. We offer competitive compensation and benefits, along with a culture that values diversity and inclusion. You will be part of a team that is at the forefront of machine learning, contributing to projects that redefine user experiences. We encourage you to apply even if your experience doesn't match every requirement, as we value curiosity and a growth mindset.