Machine Learning Engineer, Apple
Model Optimization Library, On-Device Machine Learning
$138.9-256.5k
+ Eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan
Eligible for relocation
Office located in Cupertino, CA
Personal technology company
Job no longer available
Personal technology company
1001+ employees
Job no longer available
$138.9-256.5k
+ Eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan
Eligible for relocation
Office located in Cupertino, CA
1001+ employees
Company mission
Apple's mission is to bring the best user experience to its customers through its innovative hardware, software, and services.
Role
Who you are
- 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 passionate about optimizing models for on device execution
- If you have a proven track record of applied deep learning research in model quantization, writing high quality code and shipping software, we strongly encourage you to apply
- A proven track record of applied research excellence via presentations, demos, or publications at leading ML conference or product development and deployment experience or other artifacts such as software projects
- Excellent knowledge of theory and practice of ML
- Experience in the area of model compression and quantization techniques
- Expertise with programming systems such as Python, deep learning frameworks such as PyTorch, JAX or TensorFlow
- Proven record of training, fine tuning, optimizing models
- Strong programming skills
- Excellent technical communications skills
- Demonstrated ability to design user friendly and maintainable APIs
- BS/MS/PhD in Computer Science or Electrical Engineering
- 3+ years of industry or research experience
What the job involves
- 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 small core team that collaborates heavily with researchers at Apple, ML software and hardware architecture teams and external product teams shipping state of the art optimized models on Apple devices
- We work on developing, prototyping and productizing state of the art algorithms for neural network model compression
- Our algorithms are implemented using PyTorch and optimizations are geared towards efficient deployment via Core ML
- Our APIs are available to Core ML users, both internal to Apple and external developers via the Core ML Tools optimization sub module
- Implement latest algorithms from research papers for model compression and apply to latest model architectures such as diffusion models, large language models etc
- Set up training jobs, datasets, evaluation, performance benchmarking pipelines. Applying training time and post training compression techniques
- Understanding HW capabilities and incorporating those in optimization algorithm design / enhancement
- Build upon the latest research to create new algorithms and invent new ways of applying compression to deep learning models from different domains
- Keep up with the latest AI research and collaborate with diverse teams, both internal and external to Apple, including researchers, hardware architects, and software engineers, to co develop and implement algorithms customized for Apple hardware
- Run detailed experiments and ablation studies to profile algorithms on various models, tasks, across different model sizes
- Improving model optimization documentation, writing tutorials and guides
- Self prioritize and adjust to changing priorities and asks
Our take
With a market capitalisation that reached $3 trillion in 2022, Apple is the world's most valuable company. A vertically integrated approach and obsessive focus on the user experience have helped to make Apple one of the most successful companies in the world.
Since taking over at Apple, Tim Cook has been carefully curating the company's reputation as a champion of consumer privacy, trying to set the company apart from others that monetise their users' data.
The company even went as far as to introduce sweeping privacy changes which made it difficult to tailor ads to Apple’s 1bn+ iPhone users. This rocked the $400bn digital ads market, seriously harming the revenue of its bigger rivals in the lucrative industry and forcing them to rebuild their ad infrastructure.
Despite this, the company has been shaken by supply chain issues and moved a considerable amount of its manufacturing from China to India, and increased its Vietnam presence. This is because of multi-pronged production disruption in China, the second largest customer market for Apple, and triggered an estimated 10% revenue drop in the first quarter of 2023.
Steph
Company Specialist at Welcome to the Jungle
Insights
204% employee growth in 12 months
Company
Company benefits
- Everyone has different needs and schedules, and that’s why we offer medical plans that cover both physical and mental healthcare
- In addition, employees have the option to access medical professionals from almost anywhere and get free confidential counselling — virtual or in person
- You could also be eligible for help with fitness-related expenses
- Our medical coverage plans include family-friendly features such as fertility treatments
- Because we know there’s more than one way to bring a new child into your family, we support all kinds of new parents with paid leave and our gradual return-to-work programme
- You also get paid time away if you need to care for an ill family member and free guidance to help you find support for childcare, elderly care, legal referrals and more
- At Apple, we make sure women earn the same as men performing similar work. As part of that effort, your salary history is history — we won’t ask
- Every employee here also has the opportunity to become an Apple shareholder, because all are eligible for stock grants and also a discount when purchasing Apple stock
- To help you prepare for retirement, our pension can help you meet your savings goals. And to help you prepare for the unexpected, you’ll have the security of multiple forms of income protection
- Plan on developing both professionally and personally. Apple University creates classes, seminars and beyond-the-classroom tools that help employees understand Apple’s culture, organisation, values, and role in the world. You can sharpen general business and software skills through online classes
- For more formal education related to advancing your career at Apple, we’ll reimburse you for certain educational expenses, including tuition
- Whether you’re exploring personal pursuits, welcoming a new child or caring for a family member, rest assured that you’ll have paid time away when you need it most
- When you give money to an eligible organisation, we’ll match your donations one-for-one, so your contribution is doubled. And if you choose to donate your time, we’ll contribute US$25 for every hour you volunteer. Whether you donate time or money, Apple will match your contributions up to US$10,000 a year
- Get the world’s best prices on the world’s best products — Apple products. You’ll also get deals on great third-party accessories
- And there are services to help you manage life’s day-to-day tasks: dinner reservations, pet-sitters, shopping, personal travel, financial guidance and more
Company HQ
Cupertino, CA
Leadership
Tim Cook
(CEO, not founder)Joined Apple in 1998 and held various senior roles before being promoted to CEO in 2011. Previously Spent 12 years with IBM, most recently as director of North American Fulfilment, where he led manufacturing and distribution functions for IBM’s Personal Computer Company in North and Latin America
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job