Principal Engineer, Agent Infrastructure & Memory Architecture

OKX

San Jose, United States
Base: $313,055.00 to $450,000.00; bonus/equity: pe...
On-site
Ai infrastructure engineering
Long-term memory architecture
Knowledge graph pipelines
Architect and implement long-term memory infrastructure for AI agents, including auditable, versioned, and rollback-capable middleware that supports persistent, stateful reasoning

Job Summary

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

Matching Summary

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

Salary

Base: $313,055.00 to $450,000.00; Bonus/Equity: performance bonus and long-term incentives may be provided; Benefits: full range of medical, financial, and/or other benefits

Skills & Requirements

Must-have

  • AI infrastructure engineering
  • long-term memory architecture
  • knowledge graph pipelines
  • context management systems
  • GraphRAG implementation
  • production-grade infrastructure

Nice-to-have

  • AI labs experience
  • Fintech or Web3 experience
  • startup or research environments

Key Requirements

  • 10 years of industry experience
  • PhD or Master's in Computer Science, AI, Machine Learning, Robotics or related field
  • Deep expertise in long-term memory
  • Deep expertise in knowledge graphs
  • Deep expertise in context management

Work Rights

Not specified

Tailored Resume

Cover Letter