Senior AI/ML Systems Engineer TPU & JAX Efficiency

Google is seeking a software engineer to drive TPU efficiency at scale within the Ads Training Runtime team. You will lead efforts in model stability, hardware enablement, and migration of large-scale Ads models to JAX, while optimizing across stacks for performance and cost efficiency. You will contribute to robust debugging infrastructure, support next-generation TPUs, and help deliver significant efficiency gains and business impact for Google Ads. J-18808-Ljbffr