Engineering Manager, Ml Performance

Google

Sunnyvale, CA, USA
Not specified (assumed to be hybrid based on location flexibility).
Ml training benchmarks
Tpu performance optimization
Software engineering team leadership
Google is seeking an Engineering Manager for their Machine Learning Performance team, focusing on optimizing the performance of TPUs for AI models. The role involves leadership and technical expertise to manage engineering teams, drive performance benchmarks, and collaborate on innovative algorithms

Job Summary

  • Lead a team of software engineers focused on identifying and maintaining ML training and serving benchmarks.
  • Optimize Google's custom AI chips (TPUs) for training and running massive AI/ML models.
  • Participate in algorithmic innovations exploiting new TPU hardware features and model-preserving optimizations.

Matching Summary

Match Score: 85

Google is seeking an Engineering Manager for their Machine Learning Performance team, focusing on optimizing the performance of TPUs for AI models. The role involves leadership and technical expertise to manage engineering teams, drive performance benchmarks, and collaborate on innovative algorithms.

Skills & Requirements

Must-have

  • ML training benchmarks
  • TPU performance optimization
  • software engineering team leadership
  • cross-team collaboration
  • algorithmic innovations
  • model-preserving optimizations

Nice-to-have

  • experience with open-source models
  • knowledge of compiler/runtime optimizations
  • agentic tooling experience

Key Requirements

  • Leadership in software engineering teams
  • Experience with ML benchmarks
  • Cross-functional collaboration skills

Work Rights

Not specified

Tailored Resume

Cover Letter