Senior Architect, Model Performance and Post-Training

NVIDIA Corporation; Santa Clara (CA); Presencial; USD 272.000 - 431.000; Jornada completa

Descripción de la vacante

NVIDIA Corporation is seeking a Senior Engineer to drive performance modeling, evaluation, and post-training of large language models focused on code generation and software engineering tasks. You will work at the intersection of model training, inference optimization, and empirical analysis to improve coding model capabilities and developer productivity.

You will develop meticulous frameworks to measure model performance, optimize post-training pipelines (SFT, RL, reward modeling), and build

Formación

• Bachelor’s degree in Computer Science or AI
• Excellent Python and ML fundamentals
• Proven ability to lead model development lifecycles

Responsabilidades

• Develop frameworks to measure coding model performance across tasks.
• Optimize post-training pipelines (SFT, RL, reward design).
• Analyze benchmark results and training dynamics to identify improvements.
• Improve coding model performance via targeted training interventions and error analysis.
• Build scalable infrastructure for training experiments and evaluation pipelines.

Conocimientos

Python

LLMs

Transformers

Reinforcement learning

Code generation

Experiment tracking

Dataset versioning

Educación

Bachelor's degree in CS/AI

Herramientas

Git

PyTorch

Descripción del empleo

NVIDIA is widely regarded as one of the technology industry’s most desirable employers. We lead the way in High-Performance Computing, Artificial Intelligence, and Visualization. Our core invention, the GPU, serves as the visual cortex of modern computers and powers our entire product suite. GPU deep learning ignited the modern AI era—the next great computing age with the GPU acting as the brain for everything from robots and autonomous cars to conversational AI. Today, we are known globally as "the AI computing company." We are looking to grow our teams by bringing in the smartest people in the world. Join us at the forefront of technological advancement. We are looking for a Senior Engineer to drive the performance modeling, evaluation, and post-training of large language models specialized in code generation, software engineering, and agentic coding tasks. In this role, you will work at the intersection of model training, inference optimization, and empirical analysis to improve coding model capabilities, efficiency, reliability, and real-world developer productivity.

What you’ll be doing:

• Develop meticulous frameworks to measure and analyze coding model performance across code generation, debugging, code completion, repository-level reasoning, and multi‑step software engineering tasks.
• Develop, and optimize post‑training pipelines for coding models, including supervised fine‑tuning (SFT), preference optimization, reinforcement learning (RL), and outcome‑based or process‑based reward modeling.
• Analyze benchmark results, training dynamics, failure modes, and model behaviors to find opportunities for improvement.
• Improve model performance on sophisticated coding tasks through targeted training interventions, reward design, curriculum development, and iterative error analysis.
• Build scalable, reproducible infrastructure for training experiments, evaluation pipelines, and model performance analysis.

What we need to see:

• Bachelor’s degree (or equivalent experience) in Computer Science or AI with 12+ years of meaningful work experience.
• Excellent programming skills in Python and strong understanding of large language models, transformer architectures, and the training and evaluation of generative models.
• Foundational understanding of reinforcement learning algorithms and implementation, including MDPs, policies, and reward design.
• Deep understanding of software engineering fundamentals, code generation, automated testing, and program correctness.
• Proven track record to lead model development life cycles, including dataset versioning, experiment tracking, and evaluation pipelines.

Ways to stand out from the crowd:

• Experience training or post‑training state‑of‑the‑art coding models or models designed for software engineering agents.
• Experience with execution‑based rewards, verifiable rewards, reward hacking mitigation, and automated code evaluation.
• Familiarity with coding benchmarks and evaluation environments such as SWE‑bench, etc.

Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com/ Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

The base salary range is 224,000 USD - 356,500 USD for Level5, and 272,000 USD - 431,250 USD for Level6. You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until October17, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive 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.

NVIDIA pioneered accelerated computing. Today, our AI infrastructure powers global intelligence, transforming every industry.

Learn more about NVIDIA.