Principal Engineer, Agent Infrastructure & Memory Architecture
OKX
APAC
On-site
Long-term memory infrastructure
Knowledge graph pipelines
Context management infrastructure
OKX is seeking a Principal Engineer for their Agent Infrastructure & Memory Architecture team, focusing on designing scalable AI systems with a strong emphasis on long-term memory and context management. The ideal candidate will possess advanced degrees in relevant fields and substantial industry experience, particularly in AI and machine learning
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.
Competitive total compensation package with L&D programs, education subsidy, team building, wellness allowances, and comprehensive healthcare schemes.
Matching Summary
Match Score: 85
OKX is seeking a Principal Engineer for their Agent Infrastructure & Memory Architecture team, focusing on designing scalable AI systems with a strong emphasis on long-term memory and context management. The ideal candidate will possess advanced degrees in relevant fields and substantial industry experience, particularly in AI and machine learning.
Skills & Requirements
Must-have
long-term memory infrastructure
knowledge graph pipelines
context management infrastructure
AI agent research productionization
reliability observability governance
Nice-to-have
experience in top-tier AI labs
high-frequency trading experience
Web3 or crypto platforms
startup or research-intensive 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, knowledge graphs, or context management