Senior / Staff Machine Learning Engineer, Apple Cloud AI Platform

What you'll do

• Develop training recipes, model distillation and compression methods, and training features for researchers and ML engineers.
• Build and operate scalable training frameworks and reinforcement-learning infrastructure; diagnose training instability and performance issues.
• Set technical direction across teams, mentor engineers, and ensure training runs meet privacy and security standards.

What they ask for

• 8+ years of experience in large-scale model training, including reinforcement learning, from research to production.
• 5+ years of experience building and operating large-scale model-training systems in production.
• Experience with fine-tuning and distillation techniques; hands-on experience with a higher-level training framework; deep expertise in PyTorch or JAX, including distributed training features.
• Strong programming skills in Python or C++; track record of leading technical work across teams; BS, MS, or PhD in Computer Science, Computer Engineering, or equivalent practical experience.

Nice to have

• Experience with low-precision training, large-scale parallelism, GPU/TPU architecture, or collective communication and high-speed interconnects.
• Experience profiling training workloads, building RL or post-training systems, or contributing to open source or top ML publications.

Pay

$175,000 and $308,500 USD base salary

Tools

GPUs, TPUs, PyTorch, JAX, Python, C++, Megatron-LM, DeepSpeed

AI focus

Develop and scale training for AI models, including pre-training, fine-tuning, reinforcement learning, distillation, compression, long-context, mixture-of-experts, and multimodal models.