Research Intern — Applied AI, Agentic Systems & LLMs

Axio Education is an AI-native education technology company that unifies learning, engagement, and operations through a single adaptive platform. The role involves applied AI research focused on building agentic LLM systems that incorporate various advanced AI techniques and methodologies.

Responsibilities

• You’ll work on applied AI research building agentic LLM systems that combine planning, tool use, memory, and multi-step reasoning, with additional emphasis on graph-based reasoning, knowledge graphs, representation learning, and (where useful) GNNs
• This is a hands-on role: you’ll design experiments, implement and iterate on agent workflows (strong preference for JavaScript/TypeScript), read and apply recent research papers, and evaluate reliability, cost, and real-world performance

Skills

• Strong understanding of ML fundamentals
• Strong knowledge of transformers / LLMs
• Hands-on experience building agentic systems (planning + tool use + memory + evaluation)
• Strong programming ability in JavaScript / TypeScript (Python ok, but JS/TS preferred)
• Understanding of knowledge graphs and structured representations
• Familiarity with representation learning
• Experience or academic exposure to GNNs
• Ability to read and apply research papers independently
• Self-driven, responsible, and comfortable working with minimal supervision
• Shipped or open-sourced an agent framework / agentic product in JS/TS
• Experience with RAG, long-term memory, retrieval + reranking, or graph-based retrieval
• Experience with evaluation: agent benchmarks, ablations, reliability testing, cost/perf tradeoffs
• Experience with PyTorch, JAX, or training/fine-tuning workflows
• Multi-agent systems, tool orchestration, function-calling/tool APIs, or workflow engines
• Prior research experience (academic or industry), publications, or strong technical writing

Company Overview

• Axio is an AI-native education platform that helps institutions design and deliver courses that adapt to every student. It was founded in 2023, and is headquartered in Scottsdale, Arizona, USA, with a workforce of 11-50 employees. Its website is https://axio.ai/.

Company H1B Sponsorship

• Axio Education has a track record of offering H1B sponsorships, with 1 in 2025, 1 in 2024. Please note that this does not guarantee sponsorship for this specific role.