Model Optimization Engineer
Job Summary
Job Description
This Model Optimization Engineer role focuses on optimizing deep learning models to improve user experiences. Responsibilities include implementing and productizing model optimization algorithms like quantization and pruning. The role involves collaborating with researchers, software, and hardware teams to ensure optimizations work seamlessly with the Core ML inference stack on Apple devices. The engineer will be responsible for leading testing and automation initiatives for a model optimization library, ensuring high code quality, and setting up CI pipelines. This role offers an opportunity to make a significant impact and drive the adoption of a new library.
4 skills required
Job Details
Are you excited about the impact that optimizing deep learning models can have on enabling transformative user experiences? The field of ML compression research continues to grow rapidly and new techniques to perform quantization, pruning etc are increasingly available to be ported and adopted by the ML developer community looking to ship more models in a constrained memory budget and make them run faster. We are passionate about productizing and pushing the envelope of the state of the art model optimization algorithms, to further compress and speed up the thousands of deep learning models shipping as part of Apple internal and external apps, running locally on millions of Apple devices. We work on a python library that implements a variety of training time and post training quantization algorithms and provides them to developers as simple to use, turnkey APIs, and ensures that these optimizations work seamlessly with the Core ML inference stack and Apple hardware. We are a team that collaborates heavily with researchers at Apple, ML software and hardware architecture teams and external/internal product teams shipping state of the art optimized models on Apple devices. If you are excited about making a big impact and playing a critical role in growing the user base and driving the adoption of a relatively new library, this is a great opportunity for you. We are looking for someone who is highly self motivated and looking for an opportunity to lead the testing and automation initiatives for a model optimization library for on device execution. If you are someone who is passionate about maintaining a high code quality and testability of production code, have experience setting up and maintaining CI pipelines for software projects, we strongly encourage you to apply.
Similar Jobs
Looks like we're out of matches
Set up an alert and we'll send you similar jobs the moment they appear!
Similar Skill Jobs
Looks like we're out of matches
Set up an alert and we'll send you similar jobs the moment they appear!
Jobs in Cupertino, California, United States
Looks like we're out of matches
Set up an alert and we'll send you similar jobs the moment they appear!
Similar Category Jobs
Looks like we're out of matches
Set up an alert and we'll send you similar jobs the moment they appear!
About The Company
We’re a diverse collective of thinkers and doers, continually reimagining what’s possible to help us all do what we love in new ways. And the same innovation that goes into our products also applies to our practices — strengthening our commitment to leave the world better than we found it. This is where your work can make a difference in people’s lives. Including your own.
Apple is an equal opportunity employer that is committed to inclusion and diversity. Visit apple.com/careers to learn more.
Get notified when new jobs are added by Apple
