Senior ML Engineer, AI Safety & LLM Risk

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