Research Intern — Applied AI, Agentic Systems & LLMs jobs in United States
cer-icon
Apply on Employer Site
company-logo

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
check
H1B Sponsor Likelynote

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

ML fundamentalsTransformers / LLMsJavaScript / TypeScriptAgentic systemsKnowledge graphsRepresentation learningGNNsReading research papersResponsibleMinimal supervisionSelf-driven

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

twittertwitter
company-logo
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 Stage
Total Funding
$0.1M
Key Investors
WGU Labs
2025-03-20Grant· $0.1M
2024-06-28Pre Seed

Leadership Team

leader-logo
Mark Naufel
Founder & Board Member
linkedin
Company data provided by crunchbase