OKX

Principal Engineer, Agent Infrastructure & Memory Architecture

OKX
Apply
6 months ago
Singapore, SingaporeStaff+

Responsibilities

  • Architect and implement auditable, versioned, and rollback-capable long-term memory infrastructure for AI agents
  • Design and own end-to-end GraphRAG and knowledge graph pipelines, including ingestion, schema design, indexing, retrieval optimization, and LLM integration
  • Build structured context management infrastructure such as Context Graphs for long-horizon and cross-session reasoning
  • Translate advanced agent research into maintainable, production-grade systems
  • Ensure reliability, observability, governance, traceable memory operations, evaluation frameworks, and safeguards against drift and degradation

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 at least one of long-term memory, knowledge graphs, or context management
  • Experience in top-tier AI labs or leading technology companies is preferred
  • Experience in high-frequency trading, fintech, Web3, or crypto platforms is advantageous
  • Startup or research-intensive environment experience is preferred
  • Must be comfortable operating with ambiguity and speed
  • Applicants should have current right to work in Singapore and not require visa sponsorship

Benefits

  • Learning and development programs and an education subsidy
  • Team-building programs and company events
  • Wellness and meal allowances
  • Comprehensive healthcare schemes for employees and dependants
  • Competitive total compensation package

Categories

OKX

About OKX

1,001-5,000 employees

At OKX, we believe that the future will be reshaped by Crypto, ultimately contributing to every individual's freedom. OKX began as a crypto exchange giving millions of people access to crypto trading and over time becoming among the largest platforms in the world. In recent years, we have developed one of the most connected Web3 wallets used by millions to access decentralized crypto applications (dApps). OKX is a trusted brand by hundreds of large institutions seeking access to crypto markets on a reliable platform that seamlessly connects with global banking and payments. In the last year, OKX has expanded into new markets including Australia, Brazil, Netherlands, Singapore and Turkey, with plans to launch in the US, Belgium and the UAE. We are deeply committed to shaping a fairer, more transparent and accessible society through blockchain technology. This is why we publish proof of reserves monthly, and continue to ship new innovative security features. We believe in good sportmanship and the pursuit to be better. You may see our OKX logo neatly fitted on the McLaren F1 cars, Manchester City FC team shirts, or at the Tribeca Film Festival celebrating artists and creators. Wherever you may find us, you will notice our pursuit of making our world more efficient, transparent, and connected. To learn more about OKX, download our app or visit: www.okx.com UK users follow: https://www.linkedin.com/company/okxuk/ Learn more about working at OKG, the business group developing OKX, our Digital Assets Exchange, our Web3 portal and blockchain ecosystems: bit.ly/WorkingatOKG

Contact me