Sr./Staff ML Infrastructure Engineer, Compute (TPU Scheduling) - Foundation Model

Job Summary

Summary Posted: May 08, 2026 Weekly Hours: 40 Role Number: 20... Apple is where individual imaginations gather together, committing to the values that lead to great work. Every new product we build, service we create, or Apple Store experience we deliver is the result of us making each others ideas stronger. That happens because every one of us shares a belief that we can make something wonderful and share it with the world, changing lives for the better. Its the diversity of our people and their thinking that inspires the innovation that runs through everything we do. When we bring everybody in, we can do the best work of our lives. Here, youll do more than join something youll add something!

Description

Description As a Senior/Staff Engineer on the Foundation Model Compute Infrastructure team, you will lead the design and development of scheduling and orchestration systems for large-scale TPU workloads across multi-region clusters.You will work on distributed systems that manage thousands of accelerators and enable reliable, efficient execution of large-scale training and inference jobs. This role spans scheduling algorithms, cluster lifecycle management, workload orchestration, reliability engineering, and performance optimization.