Google is seeking a Senior Software Engineer to advance ML compiler technology for TPUs and YouTube-scale recommendations. You will contribute to the Accelerated Linear Algebra (XLA) compiler, author custom kernels in JAX, and optimize models for TPU hardware across the YouTube ecosystem.
You will engage with hardware designers to co-design future processors and push performance at distributed scale, while shaping end-to-end ML software from model design to deployment.
#J-18808-Ljbffr