Member of Technical Staff – Machine Learning Engineer

About the role

Highlight is building a shared intelligence layer for modern teams, and we need a Member of Technical Staff to design and ship the AI systems that power our platform. You will work across the full ML stack, from data pipelines to model deployment, delivering features that make the product feel magical for users.

Key responsibilities

• Design, build, and optimize ML pipelines for context retrieval, action detection, and output generation.
• Implement and iterate on Retrieval‑Augmented Generation (RAG) systems, ranking models, and prompt engineering to improve quality.
• Develop comprehensive evaluation frameworks and monitoring to measure and enhance ML performance.
• Explore alternative models and fine‑tuning approaches to boost quality while reducing cost.
• Collaborate with backend and product engineering teams to ship AI‑powered features.
• Contribute to ML infrastructure decisions and best practices across the engineering organization.
• Stay current on LLM advances, retrieval techniques, and ML tooling, sharing insights with the team.

Required profile

• 4+ years of experience in ML/AI engineering with a track record of shipping production systems.
• Strong software engineering fundamentals and ability to write clean, maintainable code.
• Deep experience with retrieval, ranking, RAG, or large‑scale search systems.
• Hands‑on proficiency with PyTorch, vector databases, and modern ML infrastructure.
• Experience building LLM applications, including prompt engineering, fine‑tuning, and evaluation frameworks.
• Product‑oriented mindset, bias toward action, and ownership mentality.
• Excellent communication skills and a collaborative, team‑player attitude.

Required skills

• PyTorch
• Vector databases
• Large Language Models (LLM)
• Prompt engineering
• Fine‑tuning techniques
• Evaluation frameworks
• Retrieval systems
• Ranking models
• RAG (Retrieval‑Augmented Generation)
• Search system development