Member of Technical Staff, ML

The role

You will build AI systems across the full ML stack including data pipelines, model training, retrieval, ranking, evals, and deployment. You will own critical systems end to end and collaborate with the Head of AI Engineering to scale ML capabilities.

What you will do

• Design, build, and optimize ML pipelines for context retrieval, action detection, and output generation
• Implement and iterate on RAG systems, ranking models, and prompt engineering
• Build comprehensive evals and monitoring to measure and improve ML system performance
• Investigate alternative models and fine tuning approaches for background processing workloads
• Collaborate with Backend and Product Engineering to ship AI powered features
• Contribute to ML infrastructure decisions and best practices
• Stay current on LLM advances, retrieval techniques, and ML tooling

Required qualifications

• 4+ years in ML/AI engineering, with significant experience shipping production systems
• Strong software engineering fundamentals
• 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 (prompt engineering, fine tuning, eval frameworks)
• Product thinker
• A bias toward action
• Ownership mentality
• Excellent communication skills
• Collaborative mindset
• Must be based in or willing to relocate to SF

Nice to have

• Type-A tendencies

Benefits

• Competitive salary
• Generous equity package
• Health, dental, and vision insurance
• Flexible PTO
• Parental leave
• Paid team lunches during the week
• Relocation package

About the company

Highlight is building a shared intelligence layer for the modern workforce. It unifies context across every person and tool on your team to automate workflows.