Software Engineer - Machine Learning

FocusKPI is seeking a  Software Engineer - Machine Learning  to join one of our clients, a high-tech SaaS company.  We are looking for an experienced Machine Learning Engineer to lead the development of prompt-injection and prompt-safety models to protect the client's downstream agentic AI systems across phones, the cloud, and XR/AR.

You will design, train, and deploy classifier and guardrail models (both cloud-based and hybrid on-device) that screen agent inputs and outputs for injection attacks, unsafe content, and policy violations.

A core part of the role is post-training these models using RLHF, DPO, and related optimization techniques to push detection accuracy and false-positive rates beyond what off-the-shelf solutions can achieve.

Work Location:

•   Mountain View, CA (Onsite role, 5 days/week onsite) Duration:  12-month contract with potential to extend the contract depending on your pe...