Machine Learning Engineer, Training and Acceleration- Video Engineering

Seattle, Washington, United States
Machine Learning and AI


Weekly Hours: 40
Role Number:200251388
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. RESPONSIBILITIES: - 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.

Additional Requirements