Software Engineer III, Tensor Processing Units Compiler

Requirements:

• Bachelor’s degree or equivalent practical experience.
• 2 years of experience with software development in one or more programming languages.
• 2 years of experience with computer architecture, performance analysis, and performance modeling.
• 2 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
• 1 year of experience with ML infrastructure (e.g., model deployment, model evaluation, optimization, data processing, debugging).

Nice to haves:

• Master's degree or PhD in Computer Science or related technical fields.
• 2 years of experience with data structures and algorithms.
• Experience optimizing distributed programs at large-scale and experience with Compilers and Compiler construction.
• Experience in Machine Learning and high performance computing (HPC).
• Excellent debugging and programming concurrent/parallel computations, and working on accelerators including very long instruction word (VLIW) and Vector machines, GPUs, or digital signal processors (DSPs).

What you'll be doing:

• Write C++ code for the Tensor Processing Unit (TPU) compiler, in collaboration with Google's software development AIs.
• Carry out design reviews with peers and stakeholders to make appropriate engineering choices, write design documentation and present technical material clearly to the team.
• Improve the compiler, scaling machine learning models across hardware accelerators like TPUs, Graphics Processing Units (GPUs) inside Google and Cloud customers.
• Conduct static and runtime performance analysis of important large-scale production models.
• Design and implement performance optimizations and critical features, which increase the velocity of important production teams.

Perks and benefits:

• Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another.
• Opportunities to switch teams and projects as you and our fast-paced business grow and evolve.
• Versatility, display leadership qualities and be enthusiastic to take on new problems across the full-stack.