Machine Learning Engineer, AI Safety
$124k - $242k • Remote • US, CA, Santa Clara
Posted 13d ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
Fully remote
Categories
Machine Learning Engineer
About the job
NVIDIA is seeking a talented Machine Learning Engineer to focus on Product Security, Content Safety, and ML Fairness and Robustness for LLMs across research and production teams. This role involves assessing, quantifying, and improving the safety and inclusivity of multi-modal LLM models in a scalable manner. You will develop datasets and models for training and evaluating systems related to Content Safety, ProdSec, Robustness, and ML Fairness. The position also includes researching and implementing techniques for bias detection and mitigation in LLMs and RAG systems, defining key metrics for responsible LLM behavior, and adhering to MLOps best practices for automation, monitoring, scale, and safety. You will also contribute to the MLOps platform and develop safety tools to enhance ML team effectiveness, collaborating with cross-functional teams to address content safety and ML fairness challenges.
Responsibilities
- Develop datasets and models for training and evaluating systems for Content Safety, ProdSec, Robustness, and ML Fairness.
- Research and implement techniques for bias detection and mitigation in LLMs and RAG systems.
- Define and track key metrics for responsible LLM behavior and usage.
- Apply MLOps best 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 on content safety and ML fairness solutions.
Requirements
- Master’s or PhD in Computer Science, Electrical Engineering, or related field, or equivalent experience.
- Minimum of 2+ years of experience developing and deploying machine learning models in production.
- Strong understanding of machine learning principles and algorithms.
- Hands-on programming experience in Python and knowledge of machine learning frameworks like Keras or PyTorch.
- 1+ years of background in Content Safety, ML Fairness, Robustness, AI Model Security, or related areas.
- Experience working with large multi-modal datasets and multi-modal models.
- Good problem-solving and analytical abilities.
- Excellent collaboration and communication skills.
- Demonstrates humility, transparency, respect, and intellectual honesty.
Benefits
- Highly competitive salaries
- Comprehensive benefits package
- Equity