Senior Compiler Engineer, AI Inference Platforms

About the role

The compiler underpins inference engines used by the wider deep learning community. Work spans ahead-of-time and just-in-time builds, with an eye on speed, build time, memory use and ease of use.

At a glance

What you will do

• Study deep learning networks and write compiler optimization algorithms
• Work with framework and GPU architecture teams to speed up the next generation of deep learning software
• Define public APIs and analyze performance, building compiler techniques for AI workloads and future GPUs

What they look for

• A bachelor's, master's or PhD in computer science, computer engineering or a related field, or equal experience
• Three or more years of work or research in performance analysis and compiler optimization
• Familiarity with compiler technology such as MLIR, LLVM, XLA or Triton
• Strong C/C++ and Python skills, including debugging, performance analysis and test design

Nice to have

• CPU or GPU architecture knowledge, with CUDA or OpenCL programming
• Understanding of deep learning models and frameworks such as PyTorch or JAX
• GPU kernel authoring and profiling with tools like Nsight Compute
• Experience mentoring early-career engineers and interns

Benefits

• Equity and a benefits package
• Base salary range of 152,000 to 241,500 USD depending on location and experience