Staff ML Compute & TPU Infrastructure Engineer
appleMillbrae (CA)
Staff ML Compute & TPU Infrastructure Engineer
Posted yesterday
appleMillbrae (CA)
SENIORITY
Lead
About the role
Apple is seeking an AIML Staff ML Infrastructure Engineer to advance pre-training infrastructure in the SF Bay Area. You will work on TPUs, JAX/XLA, and distributed training to push performance and scalability across large foundation models. You will lead kernel development for attention/MoE, optimize workloads, and mentor engineers while collaborating with cross-functional teams to achieve engineering excellence.
Before you apply
Applying takes about a minute. These four things decide how fast it moves after that.
Your profile is current
It's what we read first. Occupations, seniority and locations matter more than a long history.
Two examples you can talk through
Not a portfolio — just two pieces of work where you can explain the decisions and what you'd change.
A number in mind
What you're on now and what would make you move. We negotiate better when we know both.
Your notice period
Employers plan around it, and it's the question that stalls offers most often.
Once you apply, someone reads it and calls you before anything reaches the employer — usually within two working days.