Machine Learning Engineer, Training and Acceleration- Video Engineering

Machine Learning Engineer, Training and Acceleration- Video Engineering

About the job


The Video Computer Vision org is a centralized applied research and engineering team responsible for developing real-time on-device Computer Vision and Machine Perception technologies across Apple products. We focus on a balance of research and development to deliver Apple quality, state-of-the-art experiences. Our team prides itself on innovating through the full stack, and partnering with HW, SW and ML teams to influence the sensor and silicon roadmap that brings our vision to life!

To succeed within this role, you should have shown experience in several of the following areas:

Key Qualifications

Strong experience in the area of developing machine learning training framework, or hardware acceleration of machine learning tasks

Strong software engineering skillsStrong Python programming

Strong communication skills; great work ethic and teamwork

Familiar with hardware architecture, cache utilization, data streaming model

Familiar with network optimization algorithm including quantization, sparsification, experience with knowledge distillation or NAS is a plus

Experience on computer vision and machine learning model development is a big plus

Experience on management of a team of 2-10 people is a plus


We’re looking for strong software engineer/leads to build a next generation Deep Learning technology stack to accelerate on-device machine learning capabilities and emerging innovations. You’ll be part of close nit software developers and deep learning experts working in the area of hardware aware neural network optimization, algorithms, and neural architecture search. We’re looking for candidates with strong software engineering skills, passionate about machine learning, computational science and hardware.


– Design and develop APIs for common and emerging deep learning primitives: layers, tensor operations, optimizers and more specific hardware features.

– Implement efficient tensor operations and DNN training algorithms.

– Train and evaluate DNNs for the purpose of benchmarking neural network optimization algorithms. Our framework reduces latency and power consumption of neural networks found in many Apple products.

– Perform research in emerging areas of efficient neural network development including quantization, pruning, compression and neural architecture search, as well as novel differentiable compute primitives.

– We encourage publishing novel research at top ML conferences.

Education & Experience

You’ve got ONE of the following three:

1. PhD in Machine Learning, Statistics, Optimization or related field with experience building production systems or have equivalent experience in industry

2. MS in CS with 4+ years of experience in working with large ML projects

3. BS in CS with 6+ years of experience in the industry with at least 3 years in machine learning.

Role Number: 200251388


Discover More AI Jobs:


More Information

Apply for this job Apply via Facebook
Share this job

We are one of the largest AI Communities online. Our publications have over 8.5 Million Views Annually and we have over 120K subscribers.