Staff AI Context Engineer

As a Staff AI Context Engineer specializing in RAG, Knowledge Graphs, and Memory Systems, you'll architect the information infrastructure that powers MagicSchool's AI agents. You'll design and build the knowledge organization, retrieval, and memory systems that determine what educational content our agents can access, how they navigate complex curriculum relationships, and how they maintain coherent understanding across extended teaching workflows serving millions of educators.

You will architect and implement graph-based knowledge systems(Neo4j, Neptune, etc) that represent educational content relationships, standards alignments, prerequisite chains, curriculum coherence, learning progressions, and pedagogical connections; build GraphRAG systems that combine knowledge graph traversal with vector similarity; and build robust ingestion systems that process structured and unstructured educational content, extracting entities, relationships, and metadata for knowledge base population.

Additional responsibilities for this role includes implementing NLP pipelines for educational content that extract key concepts, prerequisite relationships, learning objectives, and pedagogical metadata and partnering with Product, Research, and Educators to understand content relationships, retrieval requirements, and context needs across different teaching scenarios.