Software Engineer, TPU Compiler, PhD, Early Careers

This role is based in London, UK.

Responsibilities:

• Write product or system development code for the TPU compiler (in C++).
• Participate in or lead design reviews with peers and stakeholders.
• Contribute to a compiler that scales machine learning models across accelerators.
• Design and implement performance optimizations and critical features.
• Apply AI to the development of the Compiler.

Minimum qualifications:

• Experience with coding in data structures, algorithms, and software design.
• Research experience in AI, Distributed Systems, Machine Learning, or related fields.
• Work or educational experience in Machine Learning or AI.

Preferred qualifications:

• Currently enrolled in or graduated from a PhD program.
• Experience with parallel computing and compilers.
• Excellent debugging and programming skills for concurrent/parallel computations.