Highlightai

Highlightai

Head of AI

San Francisco office

Sponsorship not specifiedDetected 35 days ago
Vector DatabasesMachine LearningTensorFlowPyTorchData EngineeringNLPComputer VisionLLMsRAGStatisticsA/B TestingForecastingResearchLeadershipCommunicationMentoring

About the role

  • We're looking for a technical leader who's equally comfortable diving into model architectures and rallying a team around ambitious goals.
  • You thrive in ambiguity, ship fast, and hold yourself and your team to a high bar.

Responsibilities

  • Highlight is building a shared intelligence layer for the modern workforce.
  • We're hiring a Head of AI Engineering to lead and scale our ML and AI efforts as we build the engine that powers Highlight.
  • You'll own the architecture, quality, and roadmap for how we ingest, model, and surface context, starting from data pipelines and evals to the context graph that makes our system truly useful.
  • Own output quality across Highlight chat, actions, and workflows; drive toward industry leading acceptance rates.
  • Build and lead a world-class ML Engineering team; Establish AI/ML best practices across the broader engineering org.
  • This is a hands-on leadership role: you'll write code, set technical direction, mentor engineers, and collaborate closely with Backend and Product teams to ship features that feel like magic.
  • Own the context graph: architect and build Highlight's memory systems end-to-end, including data ingestion, modeling, retrieval, and evals

Requirements

  • You have deep expertise in LLMs, knowledge systems, and retrieval.
  • 7+ years in ML/AI engineering, with 3+ years leading teams or major technical initiatives
  • Experience working at top AI labs (e.g. DeepMind, OpenAI, Anthropic, Meta FAIR, xAI) or leading AI application companies (e.g. Google, Cursor, Perplexity)
  • Hands-on proficiency with modern ML stacks (PyTorch, TensorFlow, RAG, vector DBs, cloud infra)
  • You are a good fit if you have:

Compensation

  • Competitive salary and generous equity package

Benefits

  • Competitive salary and generous equity package
  • Health, dental, and vision insurance
  • Flexible PTO and parental leave
  • Research, prototype and deploy state-of-the-art machine learning text and OCR models (e.g., transformer architectures, computer vision, time series forecasting, reinforcement learning)
  • 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.
  • Stay current on knowledge representation, continual learning, and retrieval; translate insights into features that feel like magic; Research novel approaches and publish when appropriate.
Apply directly at Highlightai →Create a free account for alerts like thisView Highlightai immigration profile

This listing is sourced directly from Highlightai's careers page and normalized into a canonical job model.