Senior DL Compiler Engineer - CUDA Tile LPU
$152k - $242k • US, CA, Santa Clara
Posted 13d ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
On-site
Categories
Machine Learning Engineer
About the job
NVIDIA is seeking software engineers for the CUDA Tile team to support a new tile-based programming model for NVIDIA LPUs. This role involves designing and implementing compiler transformations, developing MLIR-based dialects and lowering passes, and optimizing the performance of tile-based kernels for various NVIDIA GPU and LPU architectures. The scope includes defining public APIs, crafting optimization techniques, and general software engineering. NVIDIA offers competitive salaries, equity, and a comprehensive benefits package, making it a desirable employer in the technology industry.
Responsibilities
- Support CUDA Tile, a new tile-based programming model for NVIDIA LPUs.
- Design and implement compiler transformations.
- Develop MLIR-based dialects and lowering passes.
- Optimize the performance of tile-based kernels across NVIDIA GPU and LPU architectures.
- Define public APIs for CUDA Tile.
- 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 a 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