Staff Software Engineer - AI Infrastructure

MagicSchool AI

full-time

Posted on: 9/17/2025

Origin: • 🇺🇸 United States

✨ AI Apply

💰 $205,000 - $240,000 per year

Lead

Distributed SystemsGoJavaScriptNext.jsNode.jsPostgresPythonTypeScript

About the role

Design and implement highly reliable agent orchestration systems, shared state management, memory systems, and multi-step workflow execution optimized for educational use cases.
Build secure, sandboxed execution environments for agent actions with robust isolation, monitoring, and safety controls.
Develop comprehensive evaluation frameworks for agent performance, safety assessments, A/B testing, and regression detection.
Create unified interfaces for multiple LLM providers, integrate with educational tool ecosystems, and build connector frameworks for curriculum data and learning management systems.
Partner with Product, Research, and Education teams to translate complex educational workflows into reliable agent capabilities from prototype to production.
Guide engineers on agent architecture patterns, evaluation methodologies, and best practices for building reliable autonomous systems.
Optimize agent execution for cost, latency, and quality while maintaining reliability standards required for educational environments.

7+ years building large-scale distributed systems with at least 2+ years in staff/senior roles
Experience with agent orchestration, workflow engines, or autonomous system infrastructure
Hands-on experience building agent applications with tool calling, context engineering, multi-step reasoning, or autonomous workflow orchestration
Familiarity with LLM integration patterns
Strong coding skills in Python, TypeScript/Node.js, or Go
Experience with our stack (TypeScript, Node.js, PostgreSQL, NextJS, Supabase) or similar technologies
Working knowledge of containers, virtualization, sandboxing, and security patterns for multi-tenant systems
Experience with metrics, observability, and incident response
Understanding of or interest in educational workflows, privacy requirements (FERPA/COPPA), and safety considerations for AI in classroom environments
Track record of architecting complex systems, mentoring engineers, and driving technical decisions