Machine Learning Engineer, AI Safety

NVIDIA is seeking a Machine Learning Engineer to work on Product Security, Content Safety, ML Fairness, and Robustness efforts for Large Language Models (LLMs). The ideal candidate will have the opportunity to tackle innovative problems in machine learning, focusing on safety for multi-modal LLMs.

The team's primary goal is to ensure that LLMs are used safely and responsibly. The successful candidate will contribute to developing datasets and models for training and evaluating Content Safety, ProdSec, Robustness, and ML Fairness.

Key responsibilities:

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

Requirements:

• Master's or PhD in Computer Science, Electrical Engineering, or a related field
• At least 2 years of experience in developing and deploying machine learning models in production
• Strong understanding of machine learning principles and algorithms
• Proficiency in Python and experience with machine learning frameworks like Keras or PyTorch
• Background in Content Safety, ML Fairness, Robustness, AI Model Security, or related areas
• Experience working with large multi-modal datasets and models
• Excellent problem-solving, analytical, collaboration, and communication skills

Preferred qualifications:

• Experience with alignment and fine-tuning of LLMs
• Knowledge of multimodal and multilingual content safety, legal, and regulatory compliance
• Familiarity with GenAI security, including prompt stability and model extraction
• Passion for AI and a demonstrated commitment to advancing the field through research and publication