Axio Education · 3 days ago
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.
Artificial Intelligence (AI)EducationInformation Technology
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
Qualification
Required
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
Preferred
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
Axio Education
Axio is an AI-native education platform that helps institutions design and deliver courses that adapt to every student.
H1B Sponsorship
Axio Education has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)
2024 (1)
Funding
Current Stage
Early StageTotal Funding
$0.1MKey Investors
WGU Labs
2025-03-20Grant· $0.1M
2024-06-28Pre Seed
Company data provided by crunchbase