Senior Machine Learning Engineer, AI Safety
NVIDIA · Remote · Santa Clara, United States · 1d ago
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.
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.