Sr./Staff ML Infrastructure Engineer, Compute (TPU Scheduling) - Foundation Model
Apple
- Location
- Cupertino, California, United States
- Job type
- Full-time
Required skills
- TPU
- GPU
- JAX
- PyTorch
- TensorFlow
- Ray
- Pathways
About the role
Experience building schedulers, resource managers, or orchestration systems for distributed workloads; experience with TPU/GPU accelerator infrastructure, distributed ML training/inference, and frameworks such as JAX, PyTorch, TensorFlow, Ray, Pathways; MS/PhD preferred.
About Apple
Designs and sells consumer electronics, software, and online services.
This page is fully interactive when JavaScript is enabled. Please enable JavaScript to apply or browse related roles.
