Role OverviewFocusKPI is seeking a Software Engineer - Machine Learning to join a high-tech SaaS company. The role involves leading the development of prompt-injection and prompt-safety models to protect the client's downstream agentic AI systems. The engineer will design, train, and deploy classifier and guardrail models to screen agent inputs and outputs for injection attacks, unsafe content, and policy violations.
What You Will Do
Design and train prompt-injection detection models and prompt-safety classifiers, build hybrid deployment pipelines, apply post-training techniques to optimize guardrail model performance, and curate and generate adversarial training data.
Why It Might Be a Fit
The ideal candidate will have 3+ years of industry experience in ML engineering or applied AI research, with demonstrated ownership of production ML systems. Strong proficiency in Python and PyTorch is required, along with hands-on experience post-training LLMs with RLHF, DPO, RLAIF, or reward modeling.
Requirements
- M.S. or Ph.D. in Computer Science, Machine Learning, Electrical Engineering, or a related field
- 3+ years of industry experience in ML engineering or applied AI research
- 2+ years of industry experience in software engineering
- Strong proficiency in Python and PyTorch
- Hands-on experience post-training LLMs with RLHF, DPO, RLAIF, or reward modeling
Benefits
- Pay Range: $95 - 110/hr
- 12-month contract with potential to extend the contract depending on performance & budget
]]>