NVIDIA is seeking a Senior AI Compiler Engineer to join their team focused on building an MLIR-based AI compiler that enhances NVIDIA's inference engine. This role involves working on performance optimizations and collaborating with framework and hardware teams to support new model patterns and GPU architectural features. The ideal candidate will contribute to software engineering tasks while ensuring efficient graph representations and optimizations for future GPU architectures. This position is integral to advancing deep learning capabilities across various applications.

• 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 performance analysis and compiler optimizations.
• Experience with compiler technologies such as MLIR, XLA, and LLVM.
• Excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and testing.
• Ability to work independently, define project goals and scope, and lead development efforts.
• Strong interpersonal skills and the ability to thrive in a dynamic, product-oriented team.

• 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, and implement compiler optimizations and kernel generation for neural networks.
• Contribute to general software engineering work.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.