Staff Software Engineer - AI Infrastructure

Job description

WHO WE ARE: MagicSchool is the premier generative AI platform for teachers. We’re just over 2 years old, and more than 6 million teachers from all over the world have joined our platform. Join a top team at a fast growing company that is working towards real social impact. Make an account and try us out at our website and connect with our passionate community on our Wall of Love.

The Role

As a Staff Software Engineer for Agents Infrastructure, you’ll architect and build the core systems that power MagicSchool’s autonomous AI capabilities. You’ll design the runtime, orchestration, and evaluation infrastructure that enables AI agents to plan, execute complex educational workflows, integrate with teaching tools, and operate safely in classroom environments.

This is a high-impact IC role where you’ll influence the technical foundation of how AI agents operate across MagicSchool’s platform, mentor engineers on agent systems, and ensure our agentic capabilities scale reliably for millions of educators.

What You’ll Do

  • Agent Runtime Architecture – Design and implement highly reliable agent orchestration systems, shared state management, memory systems, and multi-step workflow execution optimized for educational use cases.

  • Tool Integration & Safety – Build secure, sandboxed execution environments for agent actions with robust isolation, monitoring, and safety controls appropriate for educational contexts.

  • Evaluation & Reliability – Develop comprehensive evaluation frameworks for agent performance, safety assessments, A/B testing, and regression detection to ensure consistent quality in educational workflows.

  • Platform Integration – Create unified interfaces for multiple LLM providers, integrate with educational tool ecosystems, and build connector frameworks for curriculum data and learning management systems.

  • Cross-Functional Leadership – Partner with Product, Research, and Education teams to translate complex educational workflows into reliable agent capabilities from prototype to production.

  • Technical Mentorship – Guide engineers on agent architecture patterns, evaluation methodologies, and best practices for building reliable autonomous systems.

  • Performance Optimization – Optimize agent execution for cost, latency, and quality while maintaining the reliability standards required for educational environments.

What We’re Looking For

  • Deep Systems Experience – 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.

  • Agent & AI Systems – Hands-on experience building agent applications with tool calling, context engineering, multi-step reasoning, or autonomous workflow orchestration. Familiarity with LLM integration patterns.

  • Technical Stack – Strong coding skills in Python, TypeScript/Node.js, or Go. Experience with our stack (TypeScript, Node.js, PostgreSQL, NextJS, Supabase) or similar technologies.

  • Infrastructure & Security – Working knowledge of containers, virtualization, sandboxing, and security patterns for multi-tenant systems. Experience with metrics, observability, and incident response.

  • Educational Context Awareness – Understanding of or interest in educational workflows, privacy requirements (FERPA/COPPA), and safety considerations for AI in classroom environments.

  • Leadership & Impact – Track record of architecting complex systems, mentoring engineers, and driving technical decisions that balance innovation with reliability.

Nice to Have

  • Experience with agent evaluation frameworks, preference optimization, or reinforcement learning for improving agent reliability

  • Familiarity with educational technology integrations, LMS systems, or EdTech compliance requirements

  • Background in ML/AI infrastructure, model serving, or AI safety systems

  • Experience with open protocols for tool connectivity (MCP-style connectors) or browser automation

  • Security hardening experience for untrusted code execution or multi-tenant educational environments

Expectations

  • Technical Leadership – Lead the architecture and implementation of MagicSchool’s core agent infrastructure, making high-leverage decisions that enable reliable autonomous AI in education.

  • End-to-End Ownership – Own agent systems from research prototype to production deployment, ensuring they meet the reliability and safety standards required for educational use.

  • Cross-Functional Partnership – Collaborate closely with Product, Research, and Education teams to understand complex educational workflows and translate them into scalable technical solutions.

  • Mentorship & Knowledge Sharing – Coach engineers on agent architecture patterns, evaluation methodologies, and best practices for building reliable autonomous systems.

  • Innovation & Reliability – Balance cutting-edge AI capabilities with the rock-solid reliability and safety requirements of educational technology.

Why Join Us?

  • Work on cutting-edge AI technology that directly impacts educators and students.

  • Join a mission-driven team passionate about making education more efficient and equitable.

  • Flexibility of working from home, while fostering a unique culture built on relationships, trust, communication, and collaboration with our team - no matter where they live.

  • Unlimited time off to empower our employees to manage their work-life balance. We work hard for our teachers and users, and encourage our employees to rest and take the time they need.

  • Choice of employer-paid health insurance plans so that you can take care of yourself and your family. Dental and vision are also offered at very low premiums.

  • Every employee is offered generous stock options, vested over 4 years.

  • Plus a 401k match & monthly wellness stipend

Our Values:

  • Educators are Magic:  Educators are the most important ingredient in the educational process - they are the magic, not the AI. Trust them, empower them, and put them at the center of leading change in service of students and families.

  • Joy and Magic: Bring joy and magic into every learning experience - push the boundaries of what’s possible with AI.

  • Community:  Foster community that supports one another during a time of rapid technological change. Listen to them and serve their needs.

  • Innovation:  The education system is outdated and in need of innovation and change - AI is an opportunity to bring equity, access, and serve the individual needs of students better than we ever have before.

  • Responsibility: Put responsibility and safety at the forefront of the technological change that AI is bringing to education.

  • Diversity: Diversity of thought, perspectives, and backgrounds helps us serve the wide audience of educators and students around the world.

  • Excellence:  Educators and students deserve the best - and we strive for the highest quality in everything we do.

Share this job:
Please let MagicSchool AI know you found this job on Remote First Jobs 🙏

Benefits of using Remote First Jobs

Discover Hidden Jobs

Unique jobs you won't find on other job boards.

Advanced Filters

Filter by category, benefits, seniority, and more.

Priority Job Alerts

Get timely alerts for new job openings every day.

Manage Your Job Hunt

Save jobs you like and keep a simple list of your applications.

Search remote, work from home, 100% online jobs

We help you connect with top remote-first companies.

Search jobs

Hiring remote talent? Post a job

Frequently Asked Questions

What makes Remote First Jobs different from other job boards?

Unlike other job boards that only show jobs from companies that pay to post, we actively scan over 20,000 companies to find remote positions. This means you get access to thousands more jobs, including ones from companies that don't typically post on traditional job boards. Our platform is dedicated to fully remote positions, focusing on companies that have adopted remote work as their standard practice.

How often are new jobs added?

New jobs are constantly being added as our system checks company websites every day. We process thousands of jobs daily to ensure you have access to the most up-to-date remote job listings. Our algorithms scan over 20,000 different sources daily, adding jobs to the board the moment they appear.

Can I trust the job listings on Remote First Jobs?

Yes! We verify all job listings and companies to ensure they're legitimate. Our system automatically filters out spam, junk, and fake jobs to ensure you only see real remote opportunities.

Can I suggest companies to be added to your search?

Yes! We're always looking to expand our listings and appreciate suggestions from our community. If you know of companies offering remote positions that should be included in our search, please let us know. We actively work to increase our coverage of remote job opportunities.

How do I apply for jobs?

When you find a job you're interested in, simply click the 'Apply Now' button on the job listing. This will take you directly to the company's application page. We kindly ask you to mention that you found the position through Remote First Jobs when applying, as it helps us grow and improve our service 🙏

Apply