Principal Engineer Agent Infrastructure Memory Architecture

Framework Ventures

United States

Remote

USD 130,000 - 180,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Competitive total compensation package
L&D programs and Education subsidy
Various team building programs
Wellness and meal allowances
Comprehensive healthcare schemes

Job summary

A tech-focused venture capital firm is seeking an AI Infrastructure Engineer to design and build foundational systems that power next-generation AI agents. This role involves architecting long-term memory infrastructure, creating scalable knowledge graph pipelines, and transforming advanced research into production-grade systems. The ideal candidate holds a PhD or Master's in a relevant field, has 10+ years of experience, and deep expertise in areas such as long-term memory or knowledge graphs. Competitive compensation and benefits are offered.

Qualifications

  • At least 10 years of industry experience.
  • Deep expertise in long-term memory, knowledge graphs, or context management.

Responsibilities

  • Architect and implement long-term memory infrastructure for AI agents.
  • Design end-to-end GraphRAG and knowledge graph pipelines.
  • Build scalable context management infrastructure.

Skills

long-term memory
knowledge graphs
context management

Education

PhD or Master's in Computer Science, AI, Machine Learning, Robotics

Job description

About the Opportunity

We are seeking an AI Infrastructure Engineer to design and build the foundational systems that power next‑generation AI agents. This role sits at the intersection of research and production engineering, focusing on scalable long‑term memory, knowledge graph infrastructure, and robust context management systems. You will transform cutting‑edge research concepts (e.g., Context Graphs, GraphRAG, agent memory architectures) into auditable, maintainable, and production‑grade infrastructure that enables reliable, stateful AI systems.

What You’ll Be Doing

Architect and implement long‑term memory infrastructure for AI agents, including auditable, versioned, and rollback‑capable middleware that supports persistent, stateful reasoning.

Design and own end‑to‑end GraphRAG and knowledge graph pipelines, from ingestion and schema design to graph indexing, retrieval optimization, and integration with LLM‑based reasoning systems. Build scalable context management infrastructure, elevating context from prompt‑level techniques to structured, maintainable systems such as Context Graphs that enable long‑horizon and cross‑session reasoning. Translate advanced agent research into production‑grade systems, prototyping emerging architectures and establishing best practices for memory, retrieval, and infrastructure reliability. Ensure reliability, observability, and governance of agent infrastructure, including traceable memory operations, evaluation frameworks, and safeguards against drift and degradation.

What We Look For In You

PhD or Master's in Computer Science, AI, Machine Learning, Robotics or related field with at least 10 years of industry experience.

Possess a deep expertise in at least one of the following areas: long‑term memory, knowledge graphs, or context management.

Nice to Haves
  • Experience in top‑tier AI labs or Tech industry leaders
  • Experience in high‑frequency trading, fintech, Web3, or crypto platforms is a strong advantage
  • Startups or research‑intensive environments preferred; must be comfortable operating with ambiguity and speed
Perks & Benefits
  • Competitive total compensation package
  • L&D programs and Education subsidy for employees' growth and development
  • Various team building programs and company events
  • Wellness and meal allowances
  • Comprehensive healthcare schemes for employees and dependants
  • More that we love to tell you along the process!
Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Senior AI Agent Memory & Knowledge Graph Architect
Senior AI Agent Memory & Knowledge Graph Architect

Framework Ventures • United States

Remote
USD 130,000 - 180,000
Competitive total compensation package
L&D programs and Education subsidy
Various team building programs
+2
Research Scientist - AI Agent Memory Infrastructure - Global Frontier Tech Recruitment Program [...]
Research Scientist - AI Agent Memory Infrastructure - Global Frontier Tech Recruitment Program [...]

ByteDance • San Jose (CA)

On-site
USD 212,000 - 388,000
Senior AI Research Scientist / Senior AI Architect
Senior AI Research Scientist / Senior AI Architect

Platinum Venture Labs Inc • Town of Texas (WI)

On-site
USD 180,000 - 260,000
Staff Machine Learning Engineer, Agent Memory & Reasoning (University)
Staff Machine Learning Engineer, Agent Memory & Reasoning (University)

Wand AI • Palo Alto (CA)

On-site
USD 180,000 - 260,000
Staff Machine Learning Engineer, Agent Memory & Reasoning (University)
Staff Machine Learning Engineer, Agent Memory & Reasoning (University)

Wand Synthesis AI Inc • United States

On-site
USD 180,000 - 260,000
Staff Machine Learning Engineer, Agent Memory & Reasoning (University)
Staff Machine Learning Engineer, Agent Memory & Reasoning (University)

Socket.dev • United States

On-site
USD 180,000 - 240,000
Staff Machine Learning Engineer, Agent Memory & Reasoning (University)
Staff Machine Learning Engineer, Agent Memory & Reasoning (University)

Wand Inc. • Northern (KY)

Hybrid
USD 150,000 - 210,000
Software Engineer - AI Agent Memory Infrastructure San Jose Regular
Software Engineer - AI Agent Memory Infrastructure San Jose Regular

ByteDance • San Jose (CA)

On-site
USD 156,000 - 317,000
Medical, dental, and vision insurance
401(k) savings plan with company match
Paid parental leave
+2
Software Development Manager, AgentCore Memory
Software Development Manager, AgentCore Memory

Socket.dev • Jersey City (NJ)

On-site
USD 203,000 - 275,000
Health insurance
401(k) matching
Paid time off
+1
Staff Software Engineer, AI Infrastructure
Staff Software Engineer, AI Infrastructure

Harrison Clarke • San Francisco (CA)

On-site
USD 120,000 - 160,000
Competitive compensation package
Significant technical ownership
Meaningful equity