Deep Learning Compiler Engineer
$152k - $242k • US, CA, Santa Clara
Posted 1mo ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
On-site
Categories
LLM Engineer
About the job
NVIDIA is seeking software engineers for the Tensor IR & CUDA Tile team to work on compiler technologies for NVIDIA GPUs. These technologies are central to the deep learning revolution, enabling breakthroughs in generative AI, large language models, and other critical areas. You will join a top-notch team with broad impact across the deep learning community, contributing to the next era of computing where AI drives innovation.
Responsibilities
- Design and implement compiler transformations for CUDA Tile and TensorIR.
- Develop MLIR-based dialects and lowering passes.
- Optimize the performance of tile-based kernels for NVIDIA GPU architectures.
- Define public APIs for compiler technologies.
- Craft and implement compiler and optimization techniques.
- Perform performance optimization and general software engineering tasks.
Requirements
- Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, or related field (or equivalent experience).
- 3+ years of relevant work or research experience in compiler optimization, performance analysis, and IR design.
- Ability to work independently, define project goals and scope, and lead development efforts.
- 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.
Benefits
- Highly competitive salaries
- Comprehensive benefits package
- Equity
- Eligibility for benefits