I Joined Google (again)
Thu, Jul 18, 2024 in CareerJoining Google in 2016 vs. 2024. This Spring I joined Google for the second time. I’m working on core ML infrastructure, specifically ML frameworks for Keras, JAX, and Gemini. Not everyone is aware that Waymo is a completely separate sister company to Google. Going between the companies is an external hire in both directions. Since joining Waymo four years ago, I worked with a ton of excellent people on a lot of different aspects of ML infra, including quantization-aware training, ML training frameworks, and global-scale automation of TPU training and simulation pipelines.