Senior AI Compiler Engineer, MLIR

NVIDIA is an AI computing company that develops GPU-accelerated technologies for gaming, graphics, parallel computing, and artificial intelligence. The Senior AI Compiler Engineer will build an MLIR-based AI compiler for NVIDIA’s inference engine, developing graph representations, optimizations, APIs, dialects, and neural-network kernel generation while collaborating with framework and hardware teams.

Responsibilities:

• Develop MLIR-based graph representations and optimizations for future GPU architectures
• Partner with framework and hardware teams to enable new model patterns and upcoming GPU architectural features
• Define APIs and MLIR dialects, conduct performance optimizations and analysis, implement compiler optimizations and kernel generation for neural networks, and contribute to other general software engineering work