The Staff Software Engineer, TPU Performance role at Google focuses on optimizing machine learning models for Tensor Processing Units (TPUs). The position is part of the Core Machine Learning organization, which is responsible for enhancing the performance of ML workloads on TPU systems. Engineers in this role will work on projects like Gemini and collaborate with product teams to address performance challenges. This position requires a blend of technical expertise in software development and machine learning infrastructure.

• Bachelor's degree or equivalent practical experience.
• 8 years of experience in software development.
• 5 years of experience with speech/audio technology, reinforcement learning, ML infrastructure, or another ML field.
• 5 years of experience with ML design and ML infrastructure.
• 5 years of experience testing and launching software products, and 3 years of experience with software design and architecture.

• Identify and maintain ML training and serving benchmarks that are representative of Google production and the broader ML industry.
• Achieve performance for customer launches and third-party/Open-Source Software models.
• Use benchmarks to identify performance opportunities and drive improvements in collaboration with compiler and runtime teams.
• Engage with Google Product teams and researchers to solve performance problems related to new ML models and products.
• Analyze performance and efficiency metrics to identify bottlenecks and design solutions at a fleet-wide scale.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.