Head of AI Engineering — Build the Knowledge Engine

Highlight; San Francisco (CA); On-site; USD 150,000 - 200,000; Full time

Benefits offered by this job

Competitive salary and generous equity package

Health, dental, and vision insurance

Flexible PTO and parental leave

Paid team lunches and dinners during the week

Relocation package

Job summary

Highlight in San Francisco is seeking a Head of AI Engineering to lead ML and AI efforts. In this on-site role, you will own the architecture and quality of data pipelines.

The ideal candidate will have a Master's or PhD with 7+ years in ML engineering, including team leadership. You will mentor engineers and ensure high-quality outputs while leading the development of innovative AI solutions.

Key perks include competitive salary, generous equity, and flexible PTO.

Qualifications

• 7+ years in ML/AI engineering with 3+ years in leadership roles.
• Experience in shipping production AI systems at scale.
• Strong architectural experience in ML infrastructure.

Responsibilities

• Own the context graph and build memory systems end-to-end.
• Lead the full ML project lifecycle including data pipelines.
• Build and lead a world-class ML Engineering team.

Skills

Machine Learning

AI Engineering

Team Leadership

Proficiency in PyTorch

Proficiency in TensorFlow

Knowledge Graphs

Education

Master's or PhD in Computer Science or related fields

Tools

Modern ML stacks

Vector Databases

Cloud Infrastructure

Job description

Highlight in San Francisco is seeking a Head of AI Engineering to lead ML and AI efforts. In this on-site role, you will own the architecture and quality of data pipelines.

The ideal candidate will have a Master's or PhD with 7+ years in ML engineering, including team leadership. You will mentor engineers and ensure high-quality outputs while leading the development of innovative AI solutions.

Key perks include competitive salary, generous equity, and flexible PTO.