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.
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 create long-term memory, knowledge graph, GraphRAG, and context-management systems; translate emerging agent research into maintainable implementations; and ensure the reliability, observability, evaluation, and governance of these systems.
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