Machine Learning Engineer, AI Safety & Security – Supporting NVIDIA – Santa Clara (Hybrid)

About this role

NVIDIA is seeking a Machine Learning Engineer focused on AI Safety and Security for LLMs, offering a hybrid work arrangement in Santa Clara, CA. The role involves assessing and enhancing the safety and inclusivity of LLM models through various data science and engineering practices.

What you'll do

• Develop datasets for evaluating and training LLM models.
• Research bias detection and mitigation techniques for LLMs.
• Define key metrics for responsible LLM behavior.
• Perform data pre-processing and exploratory data analysis.
• Collaborate with multidisciplinary teams on machine learning solutions.
• Contribute to safety tools and repositories for ML teams.

What they're looking for

• Strong programming skills in Python
• Experience with TensorFlow or PyTorch
• Proficiency in data manipulation and visualization
• Understanding of machine learning algorithms
• Familiarity with version control systems like Git
• Problem-solving and analytical skills
• Excellent collaboration and communication abilities
• Experience with GenAI Security and Content Safety

Benefits

• Competitive pay $90/hr - $130/hr
• Full benefits
• Paid time off (PTO)
• Positive company culture
• [unknown]
• [unknown]