[Remote] Senior AI Compiler Engineer, MLIR

Note The job is a remote job and is open to candidates in USA. NVIDIA is an AI computing company developing GPU technologies that power deep learning and modern computing. The company is seeking a Senior AI Compiler Engineer to build an MLIR-based AI compiler for NVIDIA’s inference engine, focusing on performance, build efficiency, memory use, and usability across data center and edge environments. 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 Skills Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, 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 your own development efforts Strong interpersonal skills and the ability to thrive in a fast-moving, dynamic, product-oriented team Understanding of deep learning models, algorithms, and frameworks such as PyTorch and JAX Experience with GPU kernel generation targeting high performance and fast build times Proficiency in GPU architecture with CUDA or OpenCL programming experience A track record of mentoring early career engineers and interns is a bonus Benefits You will also be eligible for equity Benefits Company Overview NVIDIA designs graphics processing units and accelerated computing hardware for AI and high-performance computing applications. It was founded in 1993, and is headquartered in Santa Clara, California, USA, with a workforce of 10001+ employees. Its website is https//www.nvidia.com. Company H1B Sponsorship NVIDIA has a track record of offering H1B sponsorships, with 1247 in 2026, 1868 in 2025, 1353 in 2024, 976 in 2023, 835 in 2022, 601 in 2021, 529 in 2020. Please note that this does not guarantee sponsorship for this specific role.