This role pays more than
80%
of similar roles. Most pay
$196,375–$246,150
— the shaded band above.
At the midpoint, this role pays about
$250k
versus about
$221k
for comparable roles.
Based on 239 similar postings.
Employer
About Apple Inc
Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software
Apple Inc currently has
638 open roles
on FindRole.
Listed pay typically runs
$171,600–$272,100
across 505 roles with salary data.
Join the ML Compute team as a Staff ML Infrastructure Engineer and drive large-scale pre-training initiatives for cutting-edge foundation models, focusing on resiliency, efficiency, scalability, and resource optimization. You will enhance distributed training techniques, optimize workloads built with JAX, PyTorch, XLA, and CUDA on Kubernetes, and leverage high-performance networking technologies such as NCCL and TPU interconnects. Architect a robust MLOps platform to streamline pretraining operations, operationalize large-scale ML workloads ensuring fault-tolerance, and lead complex technical projects while mentoring engineers in your areas of expertise. Proficiency in Python or Go, distributed systems, Kubernetes, Ray, PySpark, and experience with GPUs, TPUs, and AWS Trainium is essential, along with strong communication skills to collaborate effectively across teams.
Python
Kubernetes
Docker
CI/CD
Prometheus
Grafana
AWS
GPU
CUDA
NCCL
PostgreSQL
MySQL
MongoDB
Git
Jenkins
Terraform
Ansible
Linux
Windows Server
DevOps
Scrum
Agile