Deep Learning Compiler Engineer

Job Summary

Working full-time in a remote or onsite capacity, the Deep Learning Compiler Engineer will design and implement compiler transformations, develop MLIR-based dialects and lowering passes, and optimize tile-based kernels for NVIDIA GPUs to enhance performance across various architectures.

Key responsibilities

• Design and implement compiler transformations and optimization techniques for CUDA Tile programming model
• Develop MLIR-based dialects and lowering passes to ensure efficient execution of tile-based kernels
• Define public APIs and conduct performance optimization for multiple generations of NVIDIA GPU architectures

Required qualifications

• 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
• Excellent C/C++ programming and software design skills, including debugging and performance analysis
• Ability to work independently and lead development efforts while defining project goals and scope
• Strong interpersonal skills to collaborate effectively in a dynamic product-oriented team