Do you want to help drive the progress of compilers for machine learning applications? Are you excited to learn how GPU performance powers technology such as mobile gaming, machine learning (ML), and self-driving cars? Are you passionate about challenging yourself and would you love to contribute as part of a world-class company? We are building the next generation of compiler technologies to accelerate ML workloads. We are looking for an intern to work on compiler verification & testing.

What you’ll be doing:

In this role you will work closely with deep learning compiler developers to verify new and state of the art deep learning related features and components including implementing and executing functional and performance testing and benchmarking software solutions. This would include implementing verification programs, tools, scripts, and libraries. You will apply deep learning and other sophisticated techniques to implement compiler verification solutions.

What we need to see:

• Pursuing BS/MS/PhD in Computer Science, Computer/Electrical Engineering, Mathematics or equivalent program.

• Strong Python programming skills

• Strong modern C++ programming skills

• Good understanding of machine learning domain and concepts, including Large Language Models (LLM)

Ways to stand out from the crowd:

• Knowledge of deep learning frameworks such as Pytorch, Scikit Learn, JAX/XLA or TensorRT

• Background of other programming languages and domains such as CUDA, Docker and GPU-Accelerated Cloud