Principal Engineer, Agent Infrastructure & Memory Architecture
Core
Design and build foundational systems for next-generation AI agents, focusing on scalable long-term memory, knowledge graph infrastructure, and robust context management systems.
Role type
Principal Engineer, Agent Infrastructure & Memory Architecture
Builds
Production-grade infrastructure for stateful AI reasoning, GraphRAG pipelines, and context management systems for OKX's AI agents.
Domain
Artificial Intelligence / Machine Learning / Knowledge Graphs / Web3
Deliverable
production ML models
Required skills
Long-term memory architecture, Knowledge graph design, Context management systems, GraphRAG implementation, LLM-based reasoning integration, System reliability & observability, Research-to-production translation, Middleware development, Schema design, Retrieval optimization
Preferred skills
Experience in top-tier AI labs, High-frequency trading, Fintech, Web3, or crypto platforms, Startup or research-intensive environment experience
Technologies
GraphRAG, Context Graphs, LLMs, Knowledge Graphs
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; Translate advanced agent research into production-grade systems; Ensure reliability, observability, and governance of agent infrastructure
Seniority
Principal, hands-on IC with strategy & mentorship