Design and implement scalable, reliable and high-performance machine learning infrastructure for foundation models across text, image, speech, and multi-modal domains
Collaborate with other teams to productionize state-of-the-art AI algorithms
Optimize models for performance, efficiency, and on-device intelligence
Implement machine learning systems with stringent privacy and security requirements
May also require to manage a small team of engineers.
MS or PhD in Computer Science, Machine Learning, or related technical field
Expert-level programming skills in Python
Proficiency in machine learning frameworks such as Jax, PyTorch, TensorFlow
Strong background in: Distributed training, Model optimization, and Machine learning infrastructure
Experience with large-scale model training and deployment
Experience with foundation models and large language models
Background in multi-modal AI systems
Demonstrated ability to transform research prototypes into production systems
Published research or significant contributions to open-source ML projects
Understanding of on-device machine learning techniques
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.