NVIDIA is seeking exceptional ML/DL compiler engineers to join their team focused on developing innovative technologies in machine learning compilers and AI systems. The role involves creating AI compiler solutions that enhance NVIDIA's software stack and accelerate modern machine learning models. Engineers will work closely with internal teams to push the latest advancements in AI computing. This position offers the opportunity to contribute to groundbreaking projects in a rapidly growing field.

• Bachelor's degree in Computer Science, Electrical Engineering, or a related field; MS or PhD are preferred.
• 6+ years of experience in machine learning systems development, including ML compilers and LLM inference kernels.
• Strong experience in developing or using deep learning frameworks such as PyTorch or JAX.
• Strong programming skills in Python and C/C++.

• Innovate and develop new machine learning compiler and systems technologies.
• Design, implement, and optimize compilers for high impact AI workloads.
• Build strong kernel and domain specific language solutions for state of the art kernels in LLM inference workloads.
• Develop AI-driven solutions to automate the overall development flow.
• Co-design learning system solutions with current and future ML compiler and algorithm technologies.
• Collaborate closely with other engineering teams at NVIDIA to build high impact solutions for machine learning acceleration.

• 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.