ML Engineer, LLM Safety & Security

<p><strong>Sustainable Talent&nbsp;</strong>is partnering with <strong>Nvidia</strong> a global leader who's been transforming computer graphics, PC gaming, and accelerated computing for over 25 years. We are looking for a <strong>ML Engineer, LLM Safety &amp; Security</strong>&nbsp;to support our client's team based out of in <strong>Santa Clara, CA with remote/ hybrid work options.</strong>&nbsp;</p> <p>&nbsp;</p> <p>This is a full-time (W-2) <strong>contract</strong> role. We offer competitive pay <strong>$90/hr - $130/hr </strong>based on factors like experience, education, location, etc. and provide full benefits, PTO, and amazing company culture!</p> <p>As a Machine Learning Engineer, you'll work alongside NVIDIA’s research and engineering teams, focused on AI Safety for LLMs, including multi-lingual, multi-modal, and reasoning models.&nbsp; We value expertise in data science paired with a robust data engineering foundation.&nbsp;&nbsp; This role is directed at assessing, and improving the safety and inclusivity of our LLM models in a scalable fashion.&nbsp; <strong>We seek someone proficient in programming and scripting for comprehensive data manipulation, analysis, and model fine-tuning.</strong>&nbsp; We believe in proactive problem-solving, minimal supervision, and being exceptional teammates who collaborate, think, and learn as one unit. Let's make a difference together!</p> <p><span style="text-decoration: underline;"><strong>What you’ll be doing:</strong></span></p> <ul> <li>Develop datasets and moderator models for evaluating LLM models and end-to-end systems for Content Safety, ML Fairness. These LLM models can be txt-to-txt or multimodal-to-txt.</li> <li>Develop datasets for training LLM models with SFT and RL techniques, for Content Safety, ML Fairness, Security and more.</li> <li>Research and implement cutting-edge techniques for bias detection and mitigation in LLMs and systems.</li> <li>Define and track key metrics for responsible LLM behavior and usage.</li> <li>Follow the best practices of automation, monitoring, scale, safety.</li> <li>Contribute to our repositories and develop safety tools to help ML teams be more effective.</li> <li>Data pre-processing and analysis: Collaborate with data scientists and data engineers to collect, clean, pre-process, and transform large and wide datasets.</li> <li>Conduct exploratory data analysis (EDA) to uncover insights and identify patterns that boost the model performance.</li> <li>Collaborate with multidisciplinary teams: Collaborate with product engineers, data scientists, and analysts to understand business requirements and translate them into machine learning solutions.</li> </ul> <p><span style="text-decoration: underline;"><strong>What we need to see:</strong></span></p> <ul> <li>Bachelor’s or Master’s Degree in Computer Science or related field or equivalent experience.</li> <li>2+ years of work experience as a Machine Learning Engineer or Deep Learning Scientist or a similar role, with a consistent record of successfully delivering ML solutions.</li> <li>Strong programming skills in languages such as Python. Experience with frameworks like TensorFlow, PyTorch, or scikit-learn.</li> <li>Proficiency in data manipulation, analysis, and visualization using tools like NumPy and pandas.</li> <li>Deep understanding of machine learning algorithms, statistical models, and data structures.</li> <li>Familiarity with software development practices and version control systems (e.g., Git).</li> <li>Good at problem solving and analytical ability.</li> <li>Excellent collaboration and communication skills.</li> </ul> <p><span style="text-decoration: underline;"><strong>Ways to stand out from the crowd:</strong></span></p> <ul> <li>Experience with GenAI Security including Prompt Injection Stability, Model Extraction, Confidentiality/Data Extraction, Integrity, Availability and Adversarial Robustness.</li> <li>Experience with one or more of the following areas within Content Safety: Hate/Harassment, Sexualized, Harmful/Violent, or other specific areas from your application.</li> <li>Experience with alignment/fine-tuning of LLMs - including regular LLMs as well as VLMs&nbsp; (Vision Language Model) or any-to-text</li> <li>Experience with multimodal and/or multilingual Content Safety, legal and regulatory compliance.</li> <li>Passion for AI and a demonstrated commitment to advancing the field through innovative research, prior scientific research and publication experience.</li> </ul> <p>&nbsp;</p> <p><em>Sustainable Talent is a M/F+, disabled, and veteran equal employment opportunity and affirmative action employer.</em></p>