Staff TPU Performance Architect for Large-Scale ML

Socket.dev; Sunnyvale (CA); On-site; USD 207,000 - 300,000; Full time

Job summary

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products handle information at massive scale and extend beyond web search.

We're seeking engineers who bring ideas from AI, distributed computing, and large-scale system design, with flexibility to switch teams as needed. As a software engineer, you will design, develop, test, deploy, maintain, and enhance software solutions,

Qualifications

• Bachelor's degree or equivalent practical experience.
• 8 years of software development experience.
• 5 years testing and launching software products.
• 5 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
• 3 years of experience with software design and architecture.
• Experience with ML performance analysis and benchmarking.

Responsibilities

• Focus on TPU fleet efficiency analysis and performance optimization.
• Use benchmarks to identify performance opportunities and drive improvements.
• Collaborate with product teams and researchers to deploy ML models on TPU hardware.
• Analyze performance and efficiency metrics to identify bottlenecks and implement scalable solutions.
• Explore model and data efficiency techniques such as quantization and sparsity.

Skills

Software development

Testing & launch

Performance analysis

Software architecture

ML benchmarking

Bachelor's degree or equivalence

Education

Bachelor's degree or equivalent

Master's degree or PhD

Tools

OpenXLA

GPU architectures

Job description

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products handle information at massive scale and extend beyond web search.

We're seeking engineers who bring ideas from AI, distributed computing, and large-scale system design, with flexibility to switch teams as needed. As a software engineer, you will design, develop, test, deploy, maintain, and enhance software solutions,