Senior DL Compiler Engineer -CUDA Tile

NVIDIA is a leading company in computer graphics and accelerated computing, focusing on AI innovations. They are seeking a Senior DL Compiler Engineer for the CUDA Tile team to design and implement compiler transformations and optimize performance for tile-based kernels on NVIDIA GPUs.

Responsibilities:

• Design and implement compiler transformations
• Develop MLIR-based dialects and lowering passes
• Optimize the performance of tile-based kernels to ensure they execute efficiently across multiple generations of NVIDIA GPU architectures
• Define public APIs
• Craft and implement compiler and optimization techniques
• Performance optimization and other general software engineering work

Requirements:

• Bachelors, Masters 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 your own development effort
• Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design
• Strong interpersonal skills are required along with the ability to work in a dynamic product-oriented team
• Knowledge of CPU and/or GPU architecture
• CUDA or OpenCL programming experience
• Experience with the following technologies: MLIR, LLVM, XLA, TVM and deep learning models and algorithms