NVIDIA Corporation

Machine Learning Engineer, AI Safety

NVIDIA Corporation$124K — $241K *
Technical Services
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Master's or PhD in Computer Science, Electrical Engineering, or related field, or equivalent experience.
  • 2+ years of experience in developing and deploying machine learning models in production.
  • Strong understanding of machine learning principles and algorithms.
  • Hands-on programming experience in Python with knowledge of Keras or PyTorch.
  • Experience in Content Safety or ML Fairness-related areas for at least 1 year.

Responsibilities

  • Develop datasets and models for evaluating Content Safety, ProdSec, Robustness, and ML Fairness.
  • Research and implement techniques for bias detection and mitigation in LLMs.
  • Define and track metrics for responsible LLM behavior and usage.
  • Implement MLOps best practices for automation and monitoring.
  • Contribute to the MLOps platform and tools for safety in machine learning.
  • Collaborate with teams to tackle content safety and ML fairness challenges.

Benefits

  • Equity options available.
  • Comprehensive benefits package offered.
  • Opportunity for innovative research and problem-solving.
  • Collaborative work environment with cross-functional teams.
  • Access to advanced AI tools and technologies.
Full Job Description
Our team also works in the area of safety for generative models for language, robustness, and explainability. Our LLMs are a growing area of AI products, including models and services, and we are committed to ensuring that they are used safely and responsibly. We are looking for a talented Machine Learning Engineer to work on Product Security, Content Safety, ML Fairness and Robustness efforts for LLMs across all of our research and production engineering teams. In this role, you\'ll have the opportunity to take on innovative problems in machine learning, particularly focused on safety for multi-modal LLMs. This role is directed at assessing, quantifying, and improving the safety and inclusivity of our LLM models in a scalable fashion. What you\'ll be doing: • 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 RAGs. • Define and track key metrics for responsible LLM behavior and usage. • Follow the 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. What we need to see: • 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, like 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. Ways to stand out from the crowd: • Skilled with alignment/fine-tuning of LLMs - including regular LLMs as well as VLMs (vision-language models) 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 GenAI 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. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 124,000 USD - 195,500 USD for Level 2, and 152,000 USD - 241,500 USD for Level 3. You will also be eligible for equity and benefits. Applications for this job will be accepted at least until July 28, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes.

About NVIDIA Corporation

Nvidia, a global leader in graphics, gaming, and AI technology, offers Nvidia careers and internship opportunities for those passionate about driving innovation in the tech industry. you'll find a company committed to growth, teamwork, and leadership in computer science and machine learning domains.

About Nvidia

A Pioneer in Technology and Innovation

Nvidia has cemented its reputation as a powerhouse in developing advanced graphics processing units (GPUs) and has significantly contributed to the gaming industry's evolution. Moreover, its foray into AI and machine learning has opened new frontiers in technology, making Nvidia a beacon of innovation and a desirable workplace for ambitious tech professionals.

Job Opportunities

Diverse Positions in a Dynamic Field

Nvidia is continuously on the lookout for talented individuals across various domains, including hardware and software engineering, product design, marketing, and sales. Employment opportunities at Nvidia are vast, catering to a wide range of expertise and career aspirations.

Employment in Hardware and Graphics

For those fascinated by the intricacies of hardware and graphics technology, Nvidia offers positions that sit at the forefront of gaming and computing advancements.

Growth in Machine Learning and AI

Nvidia's leadership in AI and machine learning has created numerous vacancies for specialists eager to contribute to groundbreaking projects.

Recruitment in Computer Science

With the constant demand for innovation, Nvidia's recruitment efforts focus on computer science experts capable of pushing the boundaries of what's possible.

Internship Program

Opening Doors to Future Innovators

Nvidia's internship program is designed to nurture the next generation of technology leaders, offering hands-on experience in a culture that celebrates creativity and teamwork.

Benefits and Culture

Interns at Nvidia enjoy a plethora of benefits, from competitive stipends to mentorship opportunities, all within an environment that values growth and learning.

Opportunities for Students

Whether you're an undergraduate, a master's student, or a Ph.D. candidate, Nvidia's internships provide a real-world glimpse into the tech industry, offering valuable experience in various technology fields.

Pathways to Full-Time Employment

Many interns have transitioned into full-time positions, marking the start of successful careers at Nvidia. The internship program is more than a stepping stone into the company; it’s an investment in the professional development of interns. The goal is to ensure that interns are well-equipped for future challenges.

Nvidia Careers: More Than Just a Job

Nvidia offers more than just a job to its employees; it provides a front-row seat on the journey into the future of technology. Nvidia stands as a pillar of innovation with its vast opportunities in hardware, graphics, gaming, machine learning, and computer science. Nvidia careers serve as a launching pad for talented workers who aim to redefine the technological landscape. Whether through full-time positions or internships, joining Nvidia means contributing to a legacy of breakthroughs and becoming part of a global community dedicated to pushing the boundaries of what's possible.
Learn more about NVIDIA Corporation
Size
22,473 employees
Market Cap
$350.4 billion
Industry
Net Income
$4.3 billion
Founded
1993
5 Year Trend
+31.3%
Revenue
$16.6 billion
NASDAQ

Similar Jobs

More Jobs at NVIDIA Corporation

More Technical Services Jobs

Find similar Machine Learning Engineer, AI Safety jobs: