Machine Learning Engineer, AI Safety

About the role

NVIDIA is seeking a talented Machine Learning Engineer to work on Product Security, Content Safety, ML Fairness, and Robustness efforts for Large Language Models (LLMs) across all research and production engineering teams.

Responsibilities

• Develop the datasets and models for training and evaluating models and end-to-end systems for Content Safety, ProdSec, Robustness, and ML Fairness.
• Research and implement cutting-edge techniques for bias detection and mitigation in LLMs and systems using LLMs like Retrieval-Augmented Generation (RAG).
• Define and track key metrics for responsible LLM behavior and usage.
• Follow best MLOps practices of automation, monitoring, scale, and safety.
• Contribute to the MLOps platform and develop safety tools to help ML teams be more effective.
• Collaborate with other engineers, data scientists, and researchers to develop and implement solutions to content safety and ML fairness challenges.

Requirements

• Master’s or PhD in Computer Science, Electrical Engineering or related field - or equivalent experience.
• Minimum of 2+ years of work experience in developing and deploying machine learning models in production.
• Strong understanding of machine learning principles and algorithms.
• Hands-on programming experience in Python and in-depth knowledge of machine learning frameworks, such as Keras or PyTorch.
• Background in one or more of the following broader areas for 1+ years: Content Safety, ML Fairness, Robustness, AI Model Security, or related areas.
• Experience working in a range of the following areas within Content Safety: Hate/Harassment, Sexualized, Harmful/Violent, or other specific areas from your application.
• Practice working with large multi-modal datasets and multi-modal models.
• Good at problem-solving and analytical ability.
• Excellent collaboration and communication skills.
• Demonstrates behaviors that build trust: humility, transparency, respect, and intellectual honesty.

Qualifications

• Skilled with alignment/fine-tuning of LLMs - including regular LLMs as well as Vision-Language Models (VLMs) or any-to-text.
• Proven experience with multimodal and/or multilingual content safety, legal, and regulatory compliance.
• Knowledge of robustness, including hallucinations, digressions, and generative misinformation.
• Experience with Generative AI security, including prompt stability, model extraction, confidentiality/data extraction, integrity, availability, and adversarial robustness.
• Passion for AI and a demonstrated commitment to advancing the field through innovative research, prior scientific research, and publication experience.

Benefits

Ways to stand out from the crowd include having experience with alignment/fine-tuning of LLMs, proven experience with multimodal and/or multilingual content safety, legal, and regulatory compliance, knowledge of robustness, and experience with Generative AI security.

Pay

Base salary range: $124,000 - $195,500 for Level 2, and $152,000 - $241,500 for Level 3.

Schedule

Full-time employment.

Equal Opportunity Employer

NVIDIA is an equal opportunity employer and proud to be an inclusive work environment.