Staff Software Engineer, Tpu, Performance

Google

Sunnyvale, CA, United States
Base: $207,000-$300,000; bonus/equity: included; b...
Not specified (potentially hybrid or remote)
Tensor processing units (tpu)
Jax platform optimization
Pytorch platform optimization
Google is seeking a Staff Software Engineer to join its Core Machine Learning organization, focusing on optimizing the performance of Machine Learning models using Tensor Processing Units (TPUs). The role emphasizes versatility and leadership qualities, with opportunities to work on critical projects affecting Google's flagship products

Matching Summary

Match Score: 85

Google is seeking a Staff Software Engineer to join its Core Machine Learning organization, focusing on optimizing the performance of Machine Learning models using Tensor Processing Units (TPUs). The role emphasizes versatility and leadership qualities, with opportunities to work on critical projects affecting Google's flagship products.

Salary

Base: $207,000-$300,000; Bonus/Equity: Included; Benefits: Included

Skills & Requirements

Must-have

  • Tensor Processing Units (TPU)
  • JAX platform optimization
  • PyTorch platform optimization
  • ML training and serving benchmarks
  • Performance bottleneck analysis

Nice-to-have

  • Leadership qualities
  • Full-stack versatility
  • Distributed computing expertise
  • Large-scale system design
  • Collaboration with product teams

Key Requirements

  • Software engineering experience
  • Experience with ML workloads
  • Understanding of compiler/runtime architecture

Work Rights

Not specified

Tailored Resume

Cover Letter