OKX

4 months ago

Principal Engineer, Agent Infrastructure & Memory Architecture

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI

Last checked: 4 hours agoStatus: Live

Job Summary

Lead the design and implementation of AI infrastructure powering next‑generation AI agents, focusing on scalable long‑term memory, knowledge graph infrastructure, and context management systems. Translate research concepts (Context Graphs, GraphRAG) into auditable, production‑grade systems with reliable memory, retrieval, and governance. Based in Singapore and eligible to work without visa sponsorship.

Location: Singapore

Workplace: Onsite

Employment Type: Full time

Job Function: Architecture & Urban Planning

Key Responsibilities

•Architect and implement long-term memory infrastructure for AI agents, including auditable, versioned, and rollback-capable middleware that supports persistent, stateful reasoning.
•Design and own end-to-end GraphRAG and knowledge graph pipelines, from ingestion and schema design to graph indexing, retrieval optimization, and integration with LLM-based reasoning systems.
•Build scalable context management infrastructure, elevating context from prompt-level techniques to structured, maintainable systems such as Context Graphs that enable long-horizon and cross-session reasoning.
•Translate advanced agent research into production-grade systems, prototyping emerging architectures and establishing best practices for memory, retrieval, and infrastructure reliability.
•Ensure reliability, observability, and governance of agent infrastructure, including traceable memory operations, evaluation frameworks, and safeguards against drift and degradation.

Pay and Benefits

Perks:Health InsuranceLearning BudgetWellness StipendMeal Allowances

Key Requirements

•PhD or Master’s in Computer Science, AI, Machine Learning, Robotics or related field with at least 10 years of industry experience.
•Deep expertise in long-term memory, knowledge graphs, or context management.
•Experience in top-tier AI labs or tech industry leaders is a strong plus.
•Experience in startups or research‑intensive environments and ability to operate with ambiguity and speed.
•Strong track record translating research concepts into production‑grade infrastructure.

Experience:10+ yearsAi infrastructureMemory managementKnowledge graphsGraph pipelinesLlm integration

Education:PhD / Doctorate in Computer Science

Skills:CommunicationLeadershipProblem-solving

Languages:English

Tech Stack:GraphRAGKnowledge graphsContext managementMemory infrastructureLLMs

Eligibility

Work Authorization:Authorization required. Sponsorship not provided.

Company Brief

OKX

Global cryptocurrency exchange and Web3 technology provider offering spot and derivatives trading, custody, wallet services, and blockchain infrastructure to retail and institutional users worldwide.

Industry: Trading Platforms

Company Size: Enterprise (1,001+ employees)

Growth: Established Company

Headquarters: Victoria, Seychelles

Founded: 2017

Website LinkedIn

Loading job details...

Preparing the role view and application actions.

OKX