Number of Applicants
:000+
Let AI Supercharge Your Job Hunt!
JobCopilot scans 500,000+ company career sites daily to find jobs for you
We are on a mission to reinvent how designers work in the AI era. We’re backed by top investors including First Round, Chemistry, Homebrew, Scribble and senior leaders from OpenAI, Meta, Google, Ramp, Stripe and more. We’re building the next-generation AI design tool for product teams.
We’re hiring an Lead AI Engineer to own and scale our AI infrastructure and product capabilities. You’ll build and ship production-grade, LLM-powered features, define technical direction for AI systems, and turn research into reliable, measurable product impact.
This is a hands-on role at the intersection of research, model training, and product—you’ll design the pipelines, evaluate tradeoffs, and lead execution end to end.
Own the training-to-inference pipeline for large code models—optimize inference with quantization, distillation, and caching techniques.
Research, design, and fine-tune domain-specific LLMs for code generation, refactoring, debugging, and multi-turn reasoning.
Design and optimize AI pipelines (prompting, retrieval, agents, evaluation)
Lead work on agentic systems and multi-step tool calling
Establish standards for model evaluation, benchmarking, and observability
Partner closely with product, design, and engineering to ship AI features users rely on
5+ years software engineering experience
2+ years of hands-on experience designing, training, tuning and deploying machine-learning models.
Practical experience with Natural Language Processing (NLP) and LLMs
Experience with data acquisition, data cleaning, and data pipelines
Knowledge of model optimizations (quantization, distillation, LoRA, pruning).
Experience building agentic AI systems
Ability to own systems end-to-end
Fine-tuned models for design-to-code generation
Custom SLMs for specific design workflows
A multi-step AI agent system with tool use
Real-time, streaming AI experiences
Context-aware code generation and suggestions
Salary: $300,000-$400,000 base salary
Equity: Meaningful stock options
Health Insurance: Best-in-class coverage for the employee and their entire family
Location: San Francisco HQ
Auto-Apply to Lead AI Engineer Jobs with your AI JobCopilot
Copyright © 2026 Grabjobs Pte.Ltd. All Rights Reserved.