Software Engineer - Machine Learning

FocusKPI Inc. · Mountain View, CA, United States

Location
Mountain View
Job Type
Full-time
Posted
July 06, 2026

Job Description

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 performance & budget
Pay Range: $95 - 110/hr

**No C2C resumes are co...

Ready to Apply?

Submit your application for Software Engineer - Machine Learning at FocusKPI Inc.

Apply Now