Principal Engineer, Agent Infrastructure & Memory Architecture

OKX
Hong Kong, Singapore
Full timeFunction: Architecture & Urban PlanningExperience: 10+ yearsEducation: phdSkills: ["Communication","Problem-solving","Leadership","Collaboration"]

Leading the design and build of AI infrastructure for agent memory and knowledge graphs, blending research concepts with production-grade systems. You’ll own end-to-end memory pipelines, context management, and GraphRAG, enabling reliable, long-horizon reasoning across AI agents in APAC offices.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
OKX
OKX
4 months ago

Principal Engineer, Agent Infrastructure & Memory Architecture

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 7 hours agoStatus: Live

Job Summary

Leading the design and build of AI infrastructure for agent memory and knowledge graphs, blending research concepts with production-grade systems. You’ll own end-to-end memory pipelines, context management, and GraphRAG, enabling reliable, long-horizon reasoning across AI agents in APAC offices.
Location: Hong Kong, Singapore
Employment Type: Full time
Job Function: Architecture & Urban Planning
Seniority: Sr. Manager level

Key Responsibilities

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

Pay and Benefits

Perks:Learning BudgetEducation SubsidyWellness StipendMeal AllowancesHealth Insurance

Key Requirements

  • •PhD or Master’s in Computer Science, AI, Machine Learning, Robotics or related field with at least 10 years of industry experience.
Experience:10+ yearsAIMemoryKnowledge graphsLLMContext management
Education:PhD / Doctorate
Skills:CommunicationProblem-solvingLeadershipCollaboration
Languages:English
Tech Stack:GraphRAGKnowledge graphContext managementLLMMemory architecture

Company Brief

OKX
Global cryptocurrency exchange and Web3 technology provider offering spot and derivatives trading, custody, wallet services, and blockchain infrastructure to retail and institutional users worldwide.
Industry: Trading Platforms
Company Size: Enterprise (1,001+ employees)
Growth: Established Company
Headquarters: Victoria, Seychelles
Founded: 2017
WebsiteLinkedIn