Description
NVIDIA is seeking a Deep Learning Compiler Engineering Intern to work on CUDA Tile IR. You will improve the performance and expressivity of CUDA Tile IR, analyze deep learning networks, and develop compiler optimization algorithms.
Responsibilities:
- Improve the performance and expressivity of CUDA Tile IR.
- Analyze deep learning networks and develop compiler optimization algorithms, adding new constructs to the language to better enable various workloads, and expanding support to new hardware generations.
- Collaborate with members of the deep learning software framework teams and the hardware architecture teams to accelerate the next generation of deep learning software.
Requirements:
- Currently pursuing a Bachelor's, Master's, or PhD degree within Computer Engineering, Electrical Engineering, Computer Science, or a similar field.
- Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design.
Nice to Have:
- Experience with MLIR, LLVM, XLA, deep learning models and algorithms, and deep learning framework design.
- History of contributions to open-source projects.
- Knowledge of CPU and/or GPU architecture. CUDA or OpenCL programming experience.
NVIDIA offers highly competitive salaries and a comprehensive benefits package.