Drimva

ByteDance

Software Engineer - AI Agent Memory Infrastructure

San Jose, California · Full-time · Posted Oct 28, 2026

Education
Bachelor’s degree or higher in Computer Science, Artificial Intelligence, Data Science, or a related field.

About the job

Build and operate scalable memory infrastructure that helps AI agents retain, retrieve, and use long-term, conversational, task-oriented, and multimodal context.

What you'll do

  • Design and evolve a unified memory platform and its ingestion, storage, indexing, retrieval, updating, compression, and forgetting pipelines.
  • Develop memory models and workflows for representation, retrieval and ranking, conflict resolution, summarization, multimodal fusion, and lifecycle management.
  • Work with model, application, and platform teams to productionize memory capabilities and optimize quality, latency, cost, reliability, and safety.

What they ask for

  • Bachelor’s degree or higher in Computer Science, Artificial Intelligence, Data Science, or a related field.
  • Strong experience in distributed systems, databases, information retrieval systems, or AI infrastructure, with system design and production engineering capabilities.
  • Proficiency in at least one of Go, Python, or C++; strong coding standards and engineering practices.
  • Understanding of LLM application technologies such as embeddings, retrieval-augmented generation (RAG), context engineering, retrieval systems, and long-term state management.

Nice to have

  • Experience in agent memory systems, user profiling, recommendation/search feature platforms, or knowledge base systems.
  • Contributions to or deep understanding of open-source memory frameworks such as mem0, memOS, or memU.

Tools

Go, Python, C++, LLMs, embeddings, retrieval-augmented generation (RAG), vector/graph indexing

AI focus

Build AI agent memory systems, including long-term and multimodal memory, retrieval, personalization, and lifecycle management.

Model-written summary · the original posting is authoritative

First seen Oct 28, 2026 · Last verified Nov 5, 2026 (UTC)