Senior DL Compiler Engineer -CUDA Tile
$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 CUDA Tile team to work on a new tile-based programming model for their GPUs. This role involves designing and implementing compiler transformations, developing MLIR-based dialects and lowering passes, and optimizing the performance of tile-based kernels across various NVIDIA GPU architectures. The work includes defining public APIs, crafting compiler and optimization techniques, and general software engineering tasks. NVIDIA is a leading technology company transforming computer graphics, PC gaming, and accelerated computing, now focusing on AI to define the next era of computing.
Responsibilities
- Design and implement compiler transformations for CUDA Tile.
- Develop MLIR-based dialects and lowering passes.
- Optimize the performance of tile-based kernels for NVIDIA GPU architectures.
- Define public APIs for the CUDA Tile programming model.
- Craft and implement compiler and optimization techniques.
- Perform performance optimization and general software engineering.
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
- Health insurance
- Paid time off