Software Engineer, TPU Compiler, PhD, Early Careers

Software Engineer, TPU Compiler, PhD, Early Careers Overview

As a Software Engineer focused on the TPU Compiler, you will be part of a team that develops cutting-edge technologies that transform how billions of users connect and interact with information. This role involves working on significant projects that are crucial to Google’s operations, with the flexibility to transition between teams and projects as the business evolves.

Responsibilities

• Write product or system development code for the TPU compiler using C++.
• Participate in or lead design reviews with peers and stakeholders to evaluate available technologies.
• Contribute to a compiler that scales machine learning models across TPU and GPU accelerators for internal and external clients.
• Design and implement performance optimizations and critical features to enhance the efficiency of key production teams.
• Apply AI techniques in the development of the compiler and its functionalities.

Requirements

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

Benefits

• Engage in the development of next-generation technologies that impact billions of users.
• Possibility to switch teams and projects as the business evolves.
• Work in a dynamic environment that encourages innovative ideas from various fields.

Degree Requirement: PhD

Visa Sponsorship Promising