Head of AI Engineering - ML & Knowledge Systems

Highlight AI Inc.; San Francisco (CA); On-site; USD 180,000 - 220,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

Relocation package

Job summary

Highlight AI Inc. in San Francisco is looking for a Head of AI Engineering to lead their ML and AI projects. This hands-on leadership position requires a deep understanding of LLMs and substantial experience in architecting ML systems.

The ideal candidate will have team leadership experience, a relevant Master’s or PhD, and a track record of deploying successful AI applications. This role emphasizes collaboration and a strong ownership mentality.

The position offers competitive salary, equity, and a vibrant in-office culture.

Qualifications

• 7+ years in ML/AI engineering, with 3+ years leading teams.
• Experience with knowledge graphs and retrieval systems.
• Hands-on proficiency with modern ML stacks.

Responsibilities

• Own and architect Highlight's memory systems end-to-end.
• Lead the full ML project lifecycle: data pipelines, deployment, and monitoring.
• Mentor engineers and establish AI/ML best practices.

Skills

ML/AI engineering experience

Leadership skills

LLMs expertise

Communication skills

Education

Master's or PhD in Computer Science, Machine Learning, or related

Tools

PyTorch

TensorFlow

Vector DBs

Cloud infrastructure

Job description

Highlight AI Inc. in San Francisco is looking for a Head of AI Engineering to lead their ML and AI projects. This hands-on leadership position requires a deep understanding of LLMs and substantial experience in architecting ML systems.

The ideal candidate will have team leadership experience, a relevant Master’s or PhD, and a track record of deploying successful AI applications. This role emphasizes collaboration and a strong ownership mentality.

The position offers competitive salary, equity, and a vibrant in-office culture.