Senior Machine Learning Engineer, AI Safety

NVIDIA; California; On-site; USD 184,000 - 288,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA in California is seeking Deep Learning Scientists, AI Researchers, and ML Engineers to advance AI Safety and Responsibility for Enterprise Risk Management, focusing on scaling safety for multi-modal LLMs and agentic safety.

Ideal candidates have 8+ years in ML engineering or systems software, strong Python and PyTorch skills, multimodal data experience, and excellent collaboration abilities. The role offers equity and a competitive compensation package in a fast-growing safety domain.

Qualifications

• Master’s or PhD in Computer Science, Electrical Engineering, or a related quantitative field (or equivalent experience).
• 8+ years of proven experience in systems software engineering or machine learning engineering.
• 4+ years of hands-on work experience in post-training of LLMs, including SFT, RLHF/RLAIF, safety data generation techniques, ablation studies, and production deployment.
• 1+ years of dedicated experience in at least one core safety area: LLM Security, Frontier Risks, Agentic Safety, or Multi-turn Safety Evaluation.

Responsibilities

• Evaluation: develop datasets and models to evaluate/benchmark models and end-to-end systems in core safety tracks.
• Model Pre-Training, Mid-Training, Post-Training: develop datasets and recipes for filtering training data and creating training datasets across core safety tracks.
• Model & system level techniques beyond post-training: research and deploy new approaches like Instruction Hierarchy or Risk Detection.
• Cross-Functional Collaboration: partner with engineers, data scientists, and research teams across NVIDIA to scale solutions.

Skills

Python

PyTorch

Multimodal datasets

Soft skills

Communication

Education

Master’s or PhD in Computer Science or Electrical Engineering

Tools

PyTorch

Job description

NVIDIA is seeking talented Deep Learning Scientists / AI Researchers / Machine Learning Engineers to join our rapidly growing AI Safety and Responsibility efforts for Enterprise Risk Management. In this role, you will take on innovative problems in machine learning, focusing specifically on scaling safety for multi-modal Large Language Models (LLMs) including advanced agentic safety. NVIDIA is in a unique position: we develop AI-based products across multiple domains and collaborate with the world’s leading AI companies as partners and customers. This role is directed at measuring improving the security, content safety, and inclusivity of our frontier models. Because we are expanding across multiple pillars of safety, we are looking for specialists with deep expertise in one or more of the following core focus areas:

• LLM Security: Focus on backdoors, data poisoning, latent malicious behavior, and structural model vulnerabilities.
• Frontier Risks: Focus on advanced alignment challenges, including model deception, manipulation, and loss-of-control scenarios.
• Agentic Safety: Focus on LLM-level safety for autonomous systems, including multi-turn tool-calling, orchestration, and execution risks.
• Multi-turn Safety Evaluation: Focus on robust, scalable automated evaluation methodologies for conversational and iterative multi-turn use cases.

What You'll Be Doing

• Evaluation: Develop datasets and specialized models & algorithms to evaluate/benchmark models & end-to-end systems in our core safety tracks (LLM Security, Agentic Safety, Content Safety, Hallucinations, and ML Fairness).
• Model Pre-Training, Mid-Training, Post-Training: Develop datasets and recipes for filtering training data, developing training datasets & recipes, including components like RL environments and teacher models, across our core safety tracks (LLM Security, Agentic Safety, Content Safety, Hallucinations, and ML Fairness).
• Model & system level techniques beyond post-training: Research & deploy new approaches, like Instruction Hierarchy or Risk Detection.
• Cross-Functional Collaboration: Partner with engineers, data scientists, and research teams across NVIDIA to scale solutions for LLM Security, Agentic Safety, Content Safety, Hallucinations, and ML Fairness.

What We Need To See

• Master’s or PhD in Computer Science, Electrical Engineering, or a related quantitative field (or equivalent experience).
• 8+ years of proven experience in systems software engineering or machine learning engineering.
• Post-Training Experience: 4+ years of hands-on work experience in post-training of LLMs, including Supervised Fine-Tuning (SFT), Reinforcement Learning (RLHF/RLAIF), safety data generation techniques, ablation studies, and deploying models to production.
• Core Safety Expertise: 1+ years of dedicated experience or research in at least one of the following areas:
  • LLM Security (backdoors, poisoning, latent behaviors).
  • Frontier Risks (deception, manipulation, loss-of-control).
  • Agentic Safety (LLM-level risks for multi-turn tool-calling/agents).
  • Multi-turn Safety Evaluation (dynamic and multi-turn alignment benchmarks).
  • Technical Mastery: In-depth knowledge of machine learning principles and frameworks (PyTorch preferred) with strong Python programming skills.
  • Multimodal Systems: Experience working with large multimodal datasets and multi-modal foundational models.
  • Soft Skills: Outstanding analytical problem-solving abilities paired with excellent collaboration and communication skills.
  • Cultural Alignment: Demonstrates behaviors that build trust: humility, transparency, respect, and intellectual honesty.

Ways To Stand Out From The Crowd

• Academic Track Record: Published papers on AI Safety, alignment, or machine learning security as a primary author at top-tier conferences (NeurIPS, ICML, ICLR, ACL, etc.).
• Community Contributions: Active contributions to open-source AI Safety tools, benchmarks, datasets, and/or models.
• Advanced Alignment: Proven experience with alignment/fine-tuning of Vision-Language Models (VLMs) or any-to-text foundational models.

With highly competitive salaries and a comprehensive benefits package, Nvidia is widely considered to be one of the technology industry's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working with us and our engineering teams are growing fast in some of the hottest state-of-the-art fields: Deep Learning, Artificial Intelligence, and Large Language Models. If you're a creative engineer with a real passion for robust and enjoyable user experiences, 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.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until October 10, 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.