Location
singapore
Job Type
Full-time
Posted
June 09, 2026

Job Description

About the Opportunity

We are seeking an AI Infrastructure Engineer to design and build the foundational systems that power next-generation AI agents. This role sits at the intersection of research and production engineering, focusing on scalable long-term memory , knowledge graph infrastructure , and robust context management systems .

You will transform cutting-edge research concepts (e.g., Context Graphs, GraphRAG, agent memory architectures) into auditable, maintainable, and production-grade infrastructure that enables reliable, stateful AI systems.

What You’ll Be Doing

  • 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, ...

Ready to Apply?

Submit your application for Principal Engineer, Agent Infrastructure & Memory Architecture at OKX

Apply Now