Google is seeking a software engineer to advance TPU efficiency at scale within the Ads Training Runtime team. You will drive stability, hardware enablement, and migration of large Ads models to JAX, while leading cross-stack optimizations and robust debugging to deliver major efficiency gains.
Join Google Ads to push state-of-the-art AI, collaborating with Google Research and DeepMind to solve complex challenges and deliver impact at scale.
#J-18808-LjbffrAI/ML Ads Training Engineer — TPU/JAX Expertise in mountain view at Unknown Company
This position is listed as full time and onsite.