About the job
Join our machine learning team at Apple to develop computer vision and localization technologies that help devices understand their place in the world. We're seeking a talented Machine Learning Engineer who excels at transforming research innovations into production-ready solutions, tackling the full spectrum from training large-scale models to optimizing them for real-time, on-device performance. If you're passionate about solving hard problems in spatial intelligence—from challenging real-world environments to privacy-preserving design—this is your opportunity to create experiences that millions of people use every day.
Responsibilities
Design, train, and deploy computer vision models for localization and spatial understanding on Apple devices
Build robust data pipelines and implement distributed training workflows on multi-GPU infrastructure
Optimize deep learning models for Apple's Neural Engine, achieving real-time performance while maintaining accuracy
Solve challenging problems in visual perception including place recognition, scene reconstruction, and multi-sensor fusion
Collaborate with other teams to adapt cutting-edge techniques for production constraints and real-world deployment
Validate model performance through comprehensive testing across diverse environments and use cases
Qualifications
Minimum
M.S. or Ph.D. in Computer Vision, Machine Learning, Robotics, or related field; or equivalent practical experience with a proven track record of shipping products leveraging state-of-the-art Computer Vision and Machine Learning technologies
3+ years of professional software development experience with demonstrated ability to deliver high-quality, production-ready code
Expert-level proficiency in Python and C/C++
Strong software design, problem-solving, and debugging capabilities
Hands-on experience with modern ML frameworks (PyTorch, TensorFlow/Keras, PyTorch Lightning)
Solid background with Computer Vision libraries and frameworks (OpenCV, etc.)
Experience training and optimizing machine learning models for computer vision applications
Excellent communication and collaboration skills with ability to work effectively in team environments; self-motivated and quick to adapt to new technologies
Familiarity with Gen AI coding tools. Demonstrated ability to use them for design, code generation, and debugging
Preferred
Experience developing on macOS and iOS platforms
Familiarity with Apple's internal ML/CV frameworks and tools
Experience with distributed training on clusters and multi-GPU systems
Knowledge of Apple Neural Engine optimization techniques
Experience with temporal/sequential neural network architectures (RNNs, LSTMs, Transformers for video/time-series data)
Proficiency with CMake and Xcode build systems