Back to jobs
Job Description
- Build optimizations for the latest generation of GPUs that power Google’s most critical products and services, impacting billions of users worldwide.
- Address the most challenging performance bottlenecks through Google’s unparalleled access to the latest generation of GPUs, tooling, and a decade of experience building AI accelerators.
- Drive optimizations across Google’s GPU software stack from ML compiler cost model design to optimizing high performance GPU kernels to cross node model serving configurations.
- Influence the technical direction of the GPU software ecosystem at Google by collaborating with ML, compiler design, and systems architecture teams.
- Influence the deployment of Google’s GPU fleet by working with various product teams across Google.