Large Machine Learning System Optimization Engineer
Seattle, Washington, United States
Machine Learning and AI
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
- Strong experience in the area of developing machine learning training framework, or hardware acceleration of machine learning tasks
- Strong software engineering skills in Python
- Strong communication skills; phenomenal work ethic and collaboration
- Familiar with network optimization algorithm including quantization, sparsification, experience with knowledge distillation or NAS is a plus
- Experience with hardware architecture, cache utilization, data streaming model is a plus
- Experience on large language models (LLMs), large computer vision and machine learning model, and generative AI model development is a big 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.
- 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.
Description
We’re looking for strong ML software engineer/leads to build a next generation Deep Learning technology stack to accelerate on-device large 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.
Education & Experience
BS and a minimum of 10 years relevant industry experience