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 with RLHF, DPO, and related optimization techniques to push detection accuracy and false-positive rates beyond what off-the-shelf solutions provide.

Work Location: Mountain View, CA (Onsite role, 5 days/week onsite)
Duration: 12-month contract with potential to extend the contract depending on your performance & budget
Pay Range: $95 - 110/hr

**No C2C resumes are conside...