Back to jobs
Google

Staff Engineer, TPU Co-Design

Posted 1 weeks ago

Job Description

  • Drive the definition and optimization of the hardware/software stack to enable performant training and serving of large ML models.
  • Collaborate with research and modeling teams to innovate on model architectures, focusing on scaling, quality, and their direct impact on hardware performance.
  • Lead the development of configurable architectural simulators and cycle-accurate performance models to quantify microarchitectural optimizations and evaluate architectural decisions.
  • Conduct system-level performance analysis across highly distributed ML systems, innovating new methodologies to balance compute, memory bandwidth, and inter-chip network requirements.
  • Engage with partners across hardware design, compiler development, and ML research to transition architectural innovations from concept to production.

See Your Match Score

Sign up and Renata will show you how this job matches your skills and experience.

Get Started Free
Staff Engineer, TPU Co-Design at Google | Renata