NVIDIA is seeking a Senior AI Compiler Engineer to join their team focused on developing an MLIR-based AI compiler for their inference engine. This role involves optimizing performance and usability across data center and edge environments. The engineer will collaborate with framework and hardware teams to enable new model patterns and GPU architectural features. The position is ideal for someone passionate about deep learning and compiler technologies.

• 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 fast-moving, 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 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.