San Jose, California, United States•San Jose, United States
Remote
Senior
Full Time
7 days ago
💰$313,055 - $450,000
AI InfrastructureMemory ArchitectureKnowledge GraphsContext ManagementGraphRAGLong-term MemorySenior Level
Requirements
- •PhD or Master's in Computer Science, AI, Machine Learning, Robotics or related field with at least 10 years of industry experience.
- •Possess a deep expertise in at least one of the following areas: long-term memory, knowledge graphs, or context management.
What You'll Do
- •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.
Nice to Have
- •Experience in top-tier AI labs or Tech industry leaders
- •Experience in high-frequency trading, fintech, Web3, or crypto platforms is a strong advantage
- •Startups or research-intensive environments preferred; must be comfortable operating with ambiguity and speed
Benefits
- •Performance bonus
- •Long-term incentives
- •Full range of medical, financial, and/or other benefits
About OKX
OKX is the second largest global crypto exchange by trading volume and a leading Web3 ecosystem.
Victoria, Seychelles
1000 - 5000
Blockchain & Cryptocurrency