Software Engineer · Foundation Model Infrastructure @ Apple

hi, I'mWeiwei Yang

I build large-scale ML infrastructure 🚀 for foundation model training and inference — manage TPU fleet 🛰️, workload scheduler ⚙️, and the systems that keep TPUs reliable, busy, and efficient ⚡. I am Apache Software Foundation Member, former CNCF Batch Working Group Chair, and creator of Apache YuniKorn 🦄.

Illustration of an astronaut in space

Recent Writing

All posts →
Spark Remote Shuffling Service Let the Monkey Play Annoying Flaky Tests? Setup Local Kubernetes on Mac Kubernetes Scheduler Open Hybrid Architecture Running Stateful Containers on YARN