Description
We are hiring a Senior DL Compiler Engineer to join our CUDA Tile team. As a member of this team, you will be responsible for designing and implementing compiler transformations, developing MLIR-based dialects and lowering passes, and optimizing the performance of tile-based kernels to ensure they execute efficiently across multiple generations of NVIDIA GPU architectures.
Key responsibilities include:
- Designing and implementing compiler transformations, developing MLIR-based dialects and lowering passes, and optimizing the performance of tile-based kernels
- Defining public APIs, crafting and implementing compiler and optimization techniques, performance optimization, and other general software engineering work
- Collaborating with cross-functional teams to ensure smooth integration and delivery of features
Requirements include:
- Bachelor's, Master's or Ph.D. in Computer Science, Computer Engineering or a related field (or equivalent experience)
- 3+ years of relevant work or research experience in compiler optimization, performance analysis and IR design
- Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design
- Strong interpersonal skills and ability to work in a dynamic product-oriented team
Preferred qualifications include knowledge of CPU and/or GPU architecture, CUDA or OpenCL programming experience, and experience with MLIR, LLVM, XLA, TVM and deep learning models and algorithms.
NVIDIA offers competitive salaries and a generous benefits package, making it one of the most desirable employers in the technology industry. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.