Senior AI Deep Learning Engineer

We are now looking for an AI Deep Learning Engineer! NVIDIA is seeking world-class researchers and engineers to join our deep learning team focused on building next-generation AI systems for coding. You will work at the cutting edge of large language model (LLM) development and contribute to improving AI's capabilities in understanding, generating, and interacting with code. Your work will directly impact how developers build software with the help of AI.

This role offers the unique opportunity to shape the future of programming by exploring innovations in post-training methods, agent-based workflows, and data synthesis strategies for coding LLMs.

What you'll be doing:

• Develop intelligent coding agents capable of planning, tool use, and multi-step reasoning for real-world coding workflows.
• Supervised fine-tuning and reinforcement learning to improve LLM performance on programming tasks.
• Generate and curate high-quality synthetic datasets (code, natural language, compiler traces, interaction logs) for LLM training and evaluation.
• Collaborate with cross-functional teams including CUDA teams, developer tools engineers, and foundational model researchers.
• Benchmark models on a range of code generation tasks

What we need to see:

• M.S. or Ph.D. in Computer Science, Machine Learning, or related field (or equivalent experience).
• 5+ years of experience in deep learning, with a focus on LLMs or generative models.
• Strong knowledge in one or more of the following:
• Post-training techniques for LLMs (e.g., RL, instruction tuning, alignment).
• Autonomous agents or tool-augmented LLMs.
• Synthetic data generation and scalable data pipelines for machine learning.
• Proficiency in Python, C++, and deep learning frameworks such as PyTorch
• Ability to work independently and drive projects end-to-end in a research or applied setting.
• Excellent communication and interpersonal skills to articulate complex technical concepts and collaborate effectively with cross-functional teams.

Ways to stand out from the crowd:

• Proven contributions in developing AI systems for software, hardware, or large-scale computing environments.
• Expertise in C++, CUDA, or GPU programming, with a deep understanding of chip design and computer architecture.
• Experience with GenAI and cutting-edge LLM technologies, and familiarity with the intersection of AI and hardware design.

NVIDIA is a leader in AI and GPU-accelerated computing, with a rapidly growing footprint in the datacenter business. In this role, you will be at the heart of leveraging data-driven approaches to transform hardware design and system software development. Our team collaborates across NVIDIA, engaging with deep learning research teams, CUDA Kernel developers, and Silicon Architecture groups.

What You'll Get:

• The opportunity to work alongside some of the most forward-thinking and hardworking people in the industry, shaping the future of AI.
• A creative, autonomous work environment that encourages innovation.
• The ability to influence long-term opportunities that expand NVIDIA's impact on the datacenter and beyond.

Are you ready to innovate and take on new challenges in AI at NVIDIA? We want to hear from you!

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until July 29, 2025.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.