Staff AI Engineer, Graph DB

MagicSchool is a premier generative AI platform for teachers, working towards real social impact. We're seeking a Staff AI Engineer to architect the information infrastructure that powers our AI agents, designing and building knowledge organization, retrieval, and memory systems for educational content.

Requirements

• Deep knowledge systems experience with 5+ years building large-scale information systems and 2+ years in staff/senior roles
• Graph database expertise with schema design, query optimization, and building graph-based applications
• RAG & retrieval mastery with production RAG systems, embedding selection, chunking strategies, hybrid search, reranking, and retrieval evaluation
• Embedding & NLP background with strong understanding of embedding models, fine-tuning approaches, and semantic similarity
• Technical stack with strong coding skills in Python and/or TypeScript/Node.js, experience with our stack (TypeScript, Node.js, PostgreSQL, NextJS, Supabase), and graph databases and vector stores

Benefits

• Unlimited time off
• Choice of employer-paid health insurance plans
• Generous stock options
• 401k match
• Monthly wellness stipend