Lead Machine Learning Engineer

Title: Lead Machine Learning Engineer Location: San Francisco (Hybrid) Compensation: Up to $350,000 + Equity We’re partnered with a well-funded early-stage startup rethinking how designers build in the AI era, backed by top-tier investors and operators from leading AI and technology companies. They’re building a next-generation design platform powered by LLMs, focused on helping product teams move from idea to production faster through intelligent, AI-driven workflows. This is a foundational, high-ownership role where you’ll define and scale AI systems across the product. You’ll operate at the intersection of models, systems, and product, turning cutting-edge AI capabilities into reliable, measurable user-facing features. What You’ll Do - Own AI architecture and technical direction across the product - Build and ship production-grade LLM features with clear quality and performance metrics - Design and optimize AI pipelines across prompting, retrieval, agents, and evaluation - Lead development of agentic systems and multi-step tool use workflows - Establish standards for model evaluation, benchmarking, and observability - Partner closely with product, design, and engineering to ship high-impact AI features