Requirements:

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

Nice to haves:

• 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.

What you'll be doing:

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

Perks and benefits:

• Opportunity to work on critical projects for Google.
• Potential to switch teams and projects as the business evolves.
• Contribute to cutting-edge technology in AI and ML.
• Collaborative work environment and diverse projects.