Principal Engineer, Agent Infrastructure & Memory Architecture
OKX is a leading global cryptocurrency exchange offering a wide range of trading services, including spot and derivatives trading, as well as Web3 solutions.
Maintainer signals as of 9/2/2026
Investors
Projects
About OKX
OKX is a global cryptocurrency exchange known for its trading services and financial products. The platform offers a wide range of features, including spot and derivatives trading, staking services, and a user-friendly interface suitable for both beginners and experienced traders.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
Design and build production-grade infrastructure for stateful AI agents, including long-term memory, knowledge graph, GraphRAG, and context-management systems, while ensuring reliability, observability, evaluation, and governance.
Requirements
- PhD or Master's degree in Computer Science, AI, Machine Learning, Robotics, or a related field.
- 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 and own end-to-end GraphRAG and knowledge graph pipelines.
- Build scalable context-management infrastructure for long-horizon and cross-session reasoning.
- Translate advanced agent research into production-grade systems.
- Establish best practices for memory, retrieval, and infrastructure reliability.
- Ensure reliability, observability, and governance of agent infrastructure.
- Develop traceable memory operations, evaluation frameworks, and safeguards against drift and degradation.
Benefits
- Learning and development programs
- Education subsidy
- Team-building programs and company events
- Wellness allowance
- Meal allowance
- Comprehensive healthcare schemes for employees and dependants