Staff AI / ML Engineer

📋 Description

• Design and build agentic AI systems powering learning products (multi-step reasoning, tool use).
• Select and integrate foundation models (LLMs, multimodal) balancing quality, latency, cost, and safety.
• Build evaluation and observability frameworks for AI product quality at scale (testing, human-in-the-loop, monitoring).
• Architect production-grade AI pipelines with prompt management, RAG, caching, and fallbacks.
• Collaborate with Product/Design/Engineering to translate goals into technical plans and ship features.
• Stay current with foundation models and research; apply learnings to product decisions.

🎯 Requirements

• 7+ years software engineering; 3+ years building products with LLMs/foundation models.
• Experience shipping consumer AI products with agentic patterns (tool use, multi-step reasoning, orchestration).
• Deep knowledge of foundation models: providers, prompt engineering, RAG, evaluation, and orchestration.
• Strong system/API design, production observability, and clean Python code.
• Proven track record applying new research/models to shipped products.
• Excellent collaboration and cross-functional communication.

Nice To Haves

• Experience with fine-tuning or distillation (SFT, DPO, RLHF).
• Background in edtech, consumer media, or content platforms.
• Published AI/ML research or open-source contributions.
• Experience with multi-agent systems or advanced orchestration.

🎁 Benefits

• Equity
• Medical, dental, and vision coverage
• Flexible PTO
• Comprehensive benefits package