Highlightai
Head of AI
San Francisco office. Posted 5 months ago
The role
The Head of AI Engineering will lead and scale ML and AI efforts to build the engine powering Highlight. This hands-on leadership role involves owning architecture, writing code, mentoring engineers, and collaborating with Backend and Product teams.
What you will do
- Architect and build Highlight's memory systems including data ingestion, modeling, retrieval, and evals
- Research, prototype and deploy state-of-the-art machine learning text and OCR models
- Lead the full ML project lifecycle including data pipelines, training, deployment, and monitoring
- Own output quality across Highlight chat, actions, and workflows
- Build and lead a world-class ML Engineering team
- Establish AI/ML best practices across the broader engineering org
- Stay current on knowledge representation, continual learning, and retrieval
- Research novel approaches and publish when appropriate
Required qualifications
- 7+ years in ML/AI engineering
- 3+ years leading teams or major technical initiatives
- Master’s or PhD in Computer Science, Machine Learning, Statistics, Applied Math or equivalent experience
- Strong architectural chops in designing and scaling ML infrastructure from scratch
- Hands-on proficiency with modern ML stacks (PyTorch, TensorFlow, RAG, vector DBs, cloud infra)
- Excellent communication skills
- Must be based in or willing to relocate to SF
Nice to have
- Published research (or experience shipping to production) in knowledge graphs, memory systems, retrieval, or continual learning
- Experience working at top AI labs (e.g. DeepMind, OpenAI, Anthropic, Meta FAIR, xAI) or leading AI application companies (e.g. Google, Cursor, Perplexity)
Benefits
- Competitive salary
- Generous equity package
- Health, dental, and vision insurance
- Flexible PTO
- Parental leave
- Paid team lunches and dinners 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 a team to automate workflows.