Lead AI Infrastructure & Memory Architect for Next-Gen Agents

OKX; Hong Kong; On-site; HKD 1,413,000 - 2,198,000; Full time

Benefits offered by this job

L&D programs and Education subsidy for

Wellness and meal allowances

Comprehensive healthcare schemes for

Job summary

OKX is seeking an AI Infrastructure Engineer to design foundational systems powering next-generation AI agents. The role focuses on scalable long-term memory, knowledge graph infrastructure, and robust context management.

You'll transform research concepts like Context Graphs, GraphRAG, and agent memory architectures into auditable, production-grade infrastructure that enables stateful AI systems. The role emphasizes reliability, observability, and governance of agent infra, with strong

Qualifications

• PhD or Master's in CS/AI/ML/Robotics with 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 cross-session reasoning.
• Translate research into production-grade, reliable systems with governance.

Skills

Long-term memory

Knowledge graphs

Context management

Distributed systems

Education

PhD or Master's in CS/AI/ML/Robotics

Tools

GraphRAG

Job description

OKX is seeking an AI Infrastructure Engineer to design foundational systems powering next-generation AI agents. The role focuses on scalable long-term memory, knowledge graph infrastructure, and robust context management.

You'll transform research concepts like Context Graphs, GraphRAG, and agent memory architectures into auditable, production-grade infrastructure that enables stateful AI systems. The role emphasizes reliability, observability, and governance of agent infra, with strong