Deep Learning Compiler Engineer
Core
Design and implement compiler transformations and optimizations for NVIDIA GPU kernels using CUDA Tile and TensorIR.
Role type
Senior IC deep learning compiler engineer
Builds
High-performance GPU kernels for deep learning workloads
Domain
Deep learning infrastructure / GPU computing
Deliverable
production ML models
Required skills
C/C++ programming, compiler optimization, IR design, performance analysis, MLIR dialects, lowering passes
Preferred skills
CPU/GPU architecture knowledge, CUDA/OpenCL programming, MLIR/LLVM/XLA/TVM experience
Technologies
CUDA Tile, TensorIR, MLIR, LLVM, XLA, TVM
Responsibilities
Design and implement compiler transformations, develop MLIR-based dialects and lowering passes, optimize tile-based kernel performance across GPU architectures, define public APIs, craft compiler and optimization techniques
Seniority
Senior, hands-on IC