GrepJob
OKX

Principal Engineer, Agent Infrastructure & Memory Architecture

OKX

Apply
about 3 hours ago
San Jose, CA, USA
Staff+

Base Salary

$313k - $450k/yr

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 for structured reasoning.
  • Translate advanced agent research into production-grade systems.
  • Ensure reliability and governance of agent infrastructure.

Requirements

  • PhD or Master's in Computer Science, AI, Machine Learning, Robotics, or related field.
  • At least 10 years of industry experience.
  • Deep expertise in long-term memory, knowledge graphs, or context management.

Categories

AI & MLData Engineering