Principal Engineer, Agent Infrastructure & Memory Architecture

OKX is seeking an AI Infrastructure Engineer to design and build the foundational systems that power next-generation AI agents, focusing on scalable long-term memory, knowledge graph infrastructure, and context management systems. The role transforms research concepts like Context Graphs, GraphRAG, and agent memory architectures into auditable, production-grade infrastructure.

• Architect and implement long-term memory infrastructure for AI agents, including auditable, versioned, rollback-capable middleware
• Design and own end-to-end GraphRAG and knowledge graph pipelines from ingestion to retrieval optimization
• Build scalable context management infrastructure enabling long-horizon and cross-session reasoning
• Translate advanced agent research into production-grade systems and establish best practices for memory and retrieval infrastructure

• PhD or Master's in CS, AI, ML, Robotics or related field with at least 10 years of industry experience
• Deep expertise in at least one of: long-term memory, knowledge graphs, or context management
• Experience in high-frequency trading, fintech, Web3, or crypto platforms is a strong advantage

One of the most research-heavy roles in this batch, sitting at the intersection of academic agent-memory research (Context Graphs, GraphRAG) and production infrastructure engineering for a live exchange.