Large Machine Learning System Optimization Engineer

Seattle, Washington, United States
Machine Learning and AI

Summary

Posted:
Role Number:200537436
The Video Computer Vision org is a centralized applied research and engineering team responsible for developing real-time on-device Language, 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.

Key Qualifications

  • Experience on large language models (LLMs), large computer vision and machine learning model, and generative AI model development
  • Strong software engineering skills in Python and general purpose system admin abilities. Experience with C++, Swift, or Obj-C is a plus.
  • Experimental rigor when training/evaluating DNNs for the purpose of benchmarking neural network optimization algorithms.
  • Strong communication and accountability skills; phenomenal work ethic and collaboration
  • Familiar with network optimization algorithm including quantization, sparsification, experience with knowledge distillation or NAS is a plus
  • Knowledge about hardware architecture, cache utilization, and data streaming model is a plus
  • 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.
  • Perform research in emerging areas of efficient neural network optimization
  • We encourage publishing novel research at top ML conferences.

Description

We’re looking for strong ML software engineer/leads to build a next generation Deep Learning technology stack to accelerate server-based and on-device large machine learning capabilities and emerging innovations. You’ll be part of close-knit 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.

Education & Experience

BS and a minimum of 10 years relevant industry experience

Additional Requirements

Pay & Benefits