Software Engineer, TPU Compiler

Salary: £100,000 - 100,000 per year

Requirements:

• Experience with coding in data structures, algorithms, and software design.
• Research experience in Artificial Intelligence, Distributed Systems, Machine Learning, Data Mining, Natural Language Processing, Image Classification, Spam Fighting, or related fields.
• Work or educational experience in Machine Learning or Artificial Intelligence.
• Currently enrolled in or graduated from a PhD program.
• Experience working with parallel computing.
• Experience with compilers and compiler construction.
• Excellent debugging and programming concurrent/parallel computations, and working on accelerators such as VLIW, Vector machines, GPUs, or DSPs.

Responsibilities:

• Write product or system development code for the TPU compiler in C++.
• Participate in, or lead, design reviews with peers and stakeholders to decide amongst available technologies.
• Contribute to a compiler which scales out machine learning models across accelerators like TPU/Graphics Processing Unit (GPU) at Google and Cloud.
• Design and implement performance optimizations and critical features that increase the velocity of important production teams.
• Apply AI to the development of the compiler and to the compiler itself.

Technologies:

• AI
• Cloud
• Machine Learning
• Mobile
• Security
• UX UI Design
• Web

More:

hackajob is partnering directly with Google to hire for this role. Googles software engineers develop next-generation technologies that help billions of users connect, explore, and interact with information. The TPU team develops the Accelerated Linear Algebra (XLA) TPU parallelizing compiler used to partition, optimize, and run large-scale machine learning models across multiple TPU accelerators for internal and external customers, and it is a vital part of the Google Gemini software infrastructure. Google Cloud accelerates organizations ability to digitally transform their business and industry by delivering enterprise-grade solutions that leverage Googles cutting-edge technology and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve critical business problems. We offer opportunities to work on a specific project critical to Googles needs, with the ability to switch teams and projects as our business grows and evolves.

last updated 39 week of 2026