Algorithm Optimization Engineer - ANE

Sunnyvale, California, United States
Machine Learning and AI

Summary

Posted:
Role Number:200549869
The Video Computer Vision org is a centralized applied research and engineering organization responsible for developing real-time on-device Computer Vision and Machine Perception technologies across Apple products. The team is working on exciting technologies for current and future Apple products. We balance research and product to deliver Apple quality, state-of-the-art experiences, innovating through the full stack, and partnering with HW, SW and Algorithm teams to deliver optimized solutions that brings our vision to life. We're looking for highly self-motivated and hard-working individual with expertise in optimizing machine learning and computer vision algorithms that run on ANE, GPU and CPU to join our team to deliver new technologies and experiences for Apple’s current and next generation products. Are your ready to be apart of the next big thing at Apple?

Key Qualifications

  • Rich experience with popular CNN, Transformer, and Diffusion networks and their energy-efficient variants for mobile and embedded vision applications.
  • Familiar with popular ML toolboxes (PyTorch, TensorFlow etc.) and their built in techniques for network deployment (TensorRT etc.)
  • Familiar with ML network optimization techniques, such as quantization, pruning, distillation, low-rank factorization and compression methods.
  • Bonus Qualifications:
  • Familiar with Apple ML stack (ANE, CoreML, MPS/Metal)
  • Familiar with 3D computer vision, multi-view geometry, SfM etc.

Description

Work on 3D algorithms performance optimization for Apple platforms. Analyze ML network bottleneck in the mainstream neural processor/GPU/CPU. Collaborate with ML network compiler team and hardware architecture team for both short term on-device ML model optimization and long term hardware design strategy.

Education & Experience

BS and a minimum of 10 years relevant industry experience

Additional Requirements

Pay & Benefits