Software Engineer - Machine Learning
FocusKPI, Inc.Mountain View, California, United States
Posted May 12, 2026
190000 - 220000 USD
Contract / Temporary
About the Company
FocusKPI, Inc., established in 2010, is a leading data science and technology firm offering comprehensive services including data analytics, AI-driven solutions, and staffing. Our core offerings focus on delivering actionable insights for growth and efficiency across retail media, B2B/B2C sales, and software industries. We stand out with our unique structure that competes on quality, cost, and speed, ensuring customized solutions that drive revenue lift and cost savings.
Job Description
FocusKPI is seeking a Software Engineer - Machine Learning to lead the development of prompt injection and prompt safety models to protect a high-tech SaaS company's agentic AI systems.
Requirements
- M.S. or Ph.D. in Computer Science, Machine Learning, Electrical Engineering, or a related field; or B.S. with equivalent industry experience.
- 3+ years of industry experience in ML engineering or applied AI research, with demonstrated ownership of production ML systems.
- 2+ years of industry experience in software engineering.
- Strong proficiency in Python and PyTorch (or JAX/TensorFlow), with solid software engineering fundamentals (version control, testing, and reproducible experimentation).
- Hands-on experience post-training LLMs with RLHF, DPO, RLAIF, or reward modeling, including reward design, preference data curation, and training stability.
- Hands-on experience training and deploying classifier or guardrail models for safety, content moderation, abuse detection, or adversarial robustness.
Benefits
- Competitive hourly pay
- 12-month contract with potential to extend
- Opportunity to work with a high-tech SaaS company
