This role pays more than
77%
of similar roles. Most pay
$184,562–$249,750
— the shaded band above.
At the midpoint, this role pays about
$250k
versus about
$217k
for comparable roles.
Based on 240 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.
As a Senior/Staff ML Infrastructure Engineer on the Foundation Model Compute Infrastructure team, you will lead the design and development of scheduling and orchestration systems for TPU-based workloads across multi-region clusters. Your responsibilities include building topology-aware, quota-aware, and fault-tolerant schedulers to enhance utilization and reliability while collaborating with foundation model teams to support advanced distributed training frameworks like Pathways and JAX. You will also develop automation tools for provisioning and resource management in Kubernetes environments, ensuring efficient cluster operations at scale. This role requires expertise in Python, Go, C++, Kubernetes, and large-scale cluster management systems, along with a strong background in distributed systems, scalability, and performance engineering. Experience with TPU infrastructure and distributed ML training frameworks is highly valued as you tackle the challenges of managing complex AI compute platforms.