Staff AI Engineer, Data Systems at Magicschool

About the Role

We are looking for a Staff AI Engineer specializing in RAG, Knowledge Graphs, and Memory Systems to join our team at MagicSchool. You'll architect the information infrastructure that powers our AI agents, designing and building knowledge organization, retrieval, and memory systems that determine what educational content our agents can access and how they maintain coherent understanding across extended teaching workflows.

Key Responsibilities

• →Architect and implement graph-based knowledge systems that represent educational content relationships
• →Design and evolve ontologies and schemas for educational content
• →Build GraphRAG systems that combine knowledge graph traversal with vector similarity
• →Architect and implement sophisticated retrieval-augmented generation pipelines
• →Design and operationalize embedding pipelines for educational content
• →Build robust ingestion systems that process structured and unstructured educational content
• →Invent and operationalize memory compaction mechanisms and cross-conversation memory patterns
• →Partner with Product, Research, and Educators to understand content relationships and retrieval requirements
• →Guide engineers on knowledge graph design, RAG architecture patterns, and retrieval optimization