NVIDIA is seeking an AI Research Engineer / Applied Scientist with a focus on Compilers and Low-level optimization to join their team. This role involves developing innovative technologies in machine learning compilers and AI systems, leveraging NVIDIA's software stack for comprehensive acceleration. The ideal candidate will work on AI-based technology for GPU code generation and collaborate with compiler engineers to integrate learned policies into production toolchains. This position requires a blend of research and engineering skills to transition prototypes into production-ready solutions.

• M.S. or PhD degree in Computer Engineering, Computer Science, or a related technical field.
• 5+ years of experience building AI/ML systems.
• Solid understanding of machine learning fundamentals and experimentation best practices.
• Strong software engineering skills in Python and C++.
• Hands-on experience training, fine-tuning, and post-training large models.
• Experience with reinforcement learning.
• Knowledge of prompt-engineering techniques.
• CUDA programming experience and GPU performance familiarity.

• Design and implement AI-based technology addressing core problems of low-level GPU code generation.
• Build SFT and RL training pipelines.
• Define model inputs using low-level compiler representations.
• Prototype and iterate on model architectures, prompts, and training strategies for NP-hard problems in optimizing compilers.
• Prepare datasets from compiler traces, optimization passes, and target-specific performance signals.
• Apply RL techniques to optimize for downstream objectives and run rigorous experiments, analysis, and benchmarking across workloads and hardware targets.
• Build rigorous benchmarks to assess code quality, correctness, and generation overhead.
• Partner with compiler engineers to integrate and ship learned policies with production toolchains.

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