ML Optimization and Infrastructure Engineer, Technology Development Group (TDG)

Santa Clara Valley (Cupertino), California, United States
Machine Learning and AI


Role Number:200356668
Do you want to push the limits of the best Augmented Reality platform in the world? Apple's Technology Development Group (TDG) delivers algorithms that drive revolutionary Apple products, including the augmented reality (AR) platform ARKit to create ground-breaking new products. In this position, you will have the opportunity to be part of our extraordinary team of computer vision and deep learning researchers and engineers to discover and build solutions to previously-unsolved challenges and push the state of the art in AR algorithms that will change the way people experience the world! As part of Apple’s Technology Development Group (TDG), the ML Optimization and Infrastructure Engineer will work in a dynamic team to design infrastructure to optimize ML algorithms and implement them into software frameworks. As a member of a fast-paced team, you have the unique and rewarding opportunity to work on upcoming releases of Apple products that delight and inspire millions of people every day! We invite you to contribute to our current Augmented Reality platform and craft the future of this technology. To succeed within this role, you should have shown experience in several of the following areas:

Key Qualifications

  • Experience of ML model deployment on resource constrained HW
  • Proficient in one or more ML frameworks, such as Tensorflow or Pytorch
  • Experience with ML training platform and infrastructure development
  • Strong software engineering skills in software design, development, and shipping production quality code
  • Proficiency in C/C++
  • Familiarity with Python
  • Familiarity with ML HW accelerators
  • Familiarity with software benchmarking, performance analysis


The Algorithms organization in Apple’s Technology Development Group (TDG) is looking for a skilled ML optimization and infrastructure engineer with a passion to make world-class software. In this role, you will analyze and optimize ML models to be compute and power efficient, support ML model training platform, design and implement software to efficiently execute ML algorithms for ambitious real-world problems in the domain of computer vision and computer graphics. You will collaborate closely with ML research engineers, software engineers, and ML platform, compiler, and hardware engineers to design optimal ML models and provide feedback on choice of model architectures and hardware/software requirements for efficient on-device inference.

Education & Experience

BS, MS or PhD in computer vision, machine learning, computer science, computer engineering or 2+ years of relevant industry experience with a track record of successful projects

Additional Requirements