ML Engineer: AI Safety & Fairness for LLMs (Equity)

NVIDIA Corporation; California; On-site; USD 124,000 - 242,000; Full time

Benefits offered by this job

Equity

Comprehensive benefits package

Job summary

NVIDIA Corporation is seeking a talented Machine Learning Engineer to advance content safety, ML fairness, robustness, and security for LLMs across research and production teams. You will build datasets and models for safety, reliability, and fairness, and apply cutting-edge bias detection techniques in multimodal settings.

You will define metrics, contribute to MLOps tooling, and collaborate with cross-functional teams to deploy safe, scalable AI systems.

Qualifications

• 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 ML models in production.
• Strong understanding of machine learning principles and algorithms.
• Hands-on programming experience in Python and knowledge of ML frameworks like Keras or PyTorch.
• Experience in Content Safety, ML Fairness, Robustness, AI Model Security, or related areas.
• Experience with large multimodal datasets and multimodal models.
• Excellent problem solving and analytical abilities.
• Excellent collaboration and communication skills.

Responsibilities

• Develop datasets and models for training and evaluating models for Content Safety, ProdSec, Robustness and ML Fairness.
• Research and implement techniques for bias detection and mitigation in LLMs and systems using LLMs like RAGs.
• Define and track metrics for responsible LLM behavior and usage.
• Follow best MLOps practices for automation, monitoring, scale, and safety.
• Contribute to the MLOps platform and develop safety tools for ML teams.
• Collaborate with engineers, data scientists, and researchers to address content safety and ML fairness challenges.

Skills

Python

PyTorch

Keras

ML fundamentals

Education

Master’s or PhD in CS/EE or related field

Tools

PyTorch

Keras

Job description

NVIDIA Corporation is seeking a talented Machine Learning Engineer to advance content safety, ML fairness, robustness, and security for LLMs across research and production teams. You will build datasets and models for safety, reliability, and fairness, and apply cutting-edge bias detection techniques in multimodal settings.

You will define metrics, contribute to MLOps tooling, and collaborate with cross-functional teams to deploy safe, scalable AI systems.