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