Senior ML Engineer

Hiring Company

Highlight is the shared intelligence layer that coordinates humans and AI agents across every platform.

Why Join Us

• Shared intelligence layer that captures decisions and commitments across Slack, Figma, Notion, email, and more.
• Series A startup with $40M raised, led by Khosla Ventures with General Catalyst and Valor Equity.
• Led by CEO Sergei Sorokin, former Discord VP Product who scaled Discord to 300M+ users.
• Building an intelligent operating system for teams and AI agents to eliminate the coordination tax.

Role Responsibilities

• 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 to improve quality
• Build comprehensive evals and monitoring to measure and improve ML system performance
• Investigate alternative models and fine tuning approaches for our heavy background processing workloads, implementing better systems to increase quality while improving cost efficiency
• Collaborate with Backend and Product Engineering to ship AI powered features
• Contribute to ML infrastructure decisions and best practices across the engineering org
• Stay current on LLM advances, retrieval techniques, and ML tooling; bring insights back to the team

Qualifications

• 4+ years in ML/AI engineering, with significant experience shipping production systems
• Strong software engineering fundamentals: you write clean, maintainable code
• 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: understands how to connect great ML products to constraints, evals, and modeling when building them
• A bias toward action: you ship fast, measure rigorously, and iterate based on data
• Ownership mentality (no task is too small for you) – it's all hands on deck here and we need everyone rowing the boat
• Excellent communication skills and a collaborative mindset – we're extremely lean so we're looking for a team player who works well with others
• Type-A tendencies: maybe you're extremely detail-oriented. Or maybe you take pride in shipping high-quality products. Whatever it is you do, you give it your all
• Must be based in or willing to relocate to SF – although we're flexible with days off & schedules, we have a 100% in-office culture during the week

Preferred

How can I join Jobright TNT:

1. Apply to your first Jobright TNT role
2. We review your background to determine if you meet the TNT quality bar
3. If qualified, your application is directly recommended to the employer
4. Once accepted into TNT, you may be:
5. - Invited to apply for other exclusive TNT-only roles
6. - Invited to private, invite-only hiring events with top startups

You will be notified of your TNT selection result.

PS: All Jobright TNT roles are 100% real, directly hired by top AI startups we partner with, and come with priority review and higher response rates than the normal application queue.