About the job

Location: Hybrid — Scottsdale, AZ (remote flexibility available)

Type: Internship

Hours: 40 hours/week

Compensation: $35–$45/hour

Role Overview

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 . We’re looking for someone with strong mathematical depth and real experience building agentic systems —not just prompt usage.

Required 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

Nice to Have

• 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 Description

Axio Education (Primer Global, Inc.) is an AI-native education technology company that unifies learning, engagement, and operations through a single adaptive platform. Built on a proprietary graph database and multi-agent AI infrastructure, Axio Education enables institutions to design, deliver, and manage personalized learning experiences—automating course creation, student support, and administrative workflows. Trusted by colleges and universities across the U.S., Axio Education helps its partners reduce instructional costs, improve student outcomes, and scale high-quality education globally.