Highlight
Member of Technical Staff, ML
$200K–$300K
San Francisco office · On-site · Staff · Full-time · Posted Jan 20, 2026
- Experience
- 4+ years in ML/AI engineering; no alternatives stated.
About the job
Build and operate production ML systems powering Highlight’s context retrieval, ranking, and AI-generated workflow features.
What you'll do
- Design and optimize ML pipelines for retrieval, action detection, and generation; implement RAG, ranking, and prompt engineering.
- Build evaluations and monitoring, and investigate models and fine-tuning approaches to improve quality and cost efficiency.
- Collaborate with engineering and product teams on AI features and contribute to ML infrastructure practices.
What they ask for
- 4+ years in ML/AI engineering, including significant experience shipping production systems.
- Deep experience with retrieval, ranking, RAG, or search systems at scale.
- Hands-on proficiency with PyTorch, vector databases, and modern ML infrastructure.
- Experience with LLM applications, including prompt engineering, fine-tuning, and evaluation frameworks.
Pay
$200K – $300K (base salary)
Tools
PyTorch, vector databases, RAG
AI focus
Build production AI systems across data pipelines, model training, retrieval, ranking, evaluations, and deployment; improve LLM applications and workflow automation.
Model-written summary · the original posting is authoritative