Sr Staff Machine Learning Engineer, ML Platform

Apple's ML Platform team empowers teams to develop, deploy, and operate innovative ML applications at scale. This role involves designing and building model training and fine-tuning infrastructure, defining architectures, and creating shared ML platforms for advertising. The team focuses on reliability, simplicity, and scalability, contributing to privacy-friendly advertising products.

• Design and develop model training and fine-tuning infrastructure at scale
• Build high-performing, elegant machine learning systems from the ground up
• Define and refine architectures to meet unique ad network challenges
• Create shared ML platforms, frameworks, or services used by multiple teams

• Experience building shared ML platforms, frameworks or services used by multiple teams or organizations
• Deep understanding of the ML lifecycle, including training pipelines, evaluation methodologies, and deployment patterns
• Deep understanding of deep learning architectures (Transformers, LLMs, DNNs) and training frameworks (TensorFlow, PyTorch)
• Prior experience applying ML at scale in Ads, recommender systems, information retrieval or related domains
• Prior experience in distributed training at scale and optimization techniques (model pruning, compression, quantization, distillation)