Principal Engineer Agent Infrastructure and 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 8/12/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
About the Role
You will design and build production-grade infrastructure for stateful AI agents. You will architect long-term memory middleware, own GraphRAG and knowledge graph pipelines, develop structured context-management systems, translate agent research into reliable systems, and establish observability, evaluation, and governance safeguards.
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.
- Ensure the reliability, observability, and governance of agent infrastructure.
Benefits
- Competitive total compensation package
- Learning and development programs
- Education subsidy
- Team-building programs and company events
- Wellness and meal allowances
- Comprehensive healthcare schemes for employees and dependants