AI Memory Infrastructure Tech Lead

ByteDance

San Jose (CA)

On-site

USD 244,800 - 588,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Medical/Dental/Vision insurance
401(k) matching
Parental leave
Disability coverage
Life insurance
Wellbeing benefits
Holidays
Sick days
Personal time

Job summary

ByteDance is hiring for Tech Lead, Software Engineer - AI Agent Memory Infrastructure in San Jose. You will help design and operate memory infrastructure for long-term and conversational AI, enabling personalized and context-aware experiences.

You will collaborate with model, product, and platform teams to productionize memory capabilities, optimize performance, latency, and cost, while exploring novel memory representations and multimodal data support.

Qualifications

  • Bachelor’s degree or higher in CS/AI/DS or related fields.
  • Strong experience in distributed systems, databases, or AI infrastructure.
  • Proficient in Go, Python, or C++ with coding best practices.

Responsibilities

  • Design and evolve memory infrastructure for AI agents and long-term memory.
  • Architect pipelines for low-latency, high-availability memory systems.
  • Tackle memory representation, retrieval, ranking, and fusion challenges.
  • Collaborate with model/product/platform teams to productionize memory capabilities.

Skills

Distributed systems
Databases
Information retrieval
Go
Python
C++
LLM embeddings
RAG
Context engineering
Memory systems

Education

Bachelor’s degree or higher in Computer Science, Artificial Intelligence, Data Science, or related fields

Tools

mem0
memOS
memU

Job description

ByteDance is hiring for Tech Lead, Software Engineer - AI Agent Memory Infrastructure in San Jose. You will help design and operate memory infrastructure for long-term and conversational AI, enabling personalized and context-aware experiences.

You will collaborate with model, product, and platform teams to productionize memory capabilities, optimize performance, latency, and cost, while exploring novel memory representations and multimodal data support.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

AI Agent Memory Systems Research Scientist
AI Agent Memory Systems Research Scientist

ByteDance • San Jose (CA)

On-site
USD 212,000 - 388,000
Senior AI Systems Scientist - Memory & Scale
Senior AI Systems Scientist - Memory & Scale

ByteDance • San Jose (CA)

On-site
USD 212,800 - 387,600
Medical Insurance
Dental Insurance
Vision Insurance
+9
AI Agent Memory Architect — Low-Latency Infrastructure
AI Agent Memory Architect — Low-Latency Infrastructure

ByteDance • San Jose (CA)

On-site
USD 156,000 - 317,000
Medical, dental, and vision insurance
401(k) savings plan with company match
Paid parental leave
+2
Memory Systems Research Scientist (AI Agents)
Memory Systems Research Scientist (AI Agents)

ByteDance • San Jose (CA)

On-site
USD 156,000 - 388,000
Medical, dental, and vision insurance
401(k) savings plan with company match
Paid parental leave
+1
Software Engineer - AI Agent Memory Infrastructure San Jose Regular
Software Engineer - AI Agent Memory Infrastructure San Jose Regular

ByteDance • San Jose (CA)

On-site
USD 156,000 - 317,000
Medical, dental, and vision insurance
401(k) savings plan with company match
Paid parental leave
+2
Tech Lead, Software Engineer - AI Agent Memory Infrastructure
Tech Lead, Software Engineer - AI Agent Memory Infrastructure

ByteDance • San Jose (CA)

On-site
USD 244,800 - 588,000
Medical/Dental/Vision insurance
401(k) matching
Parental leave
+6
Lead Engineer, AI Compute Infrastructure
Lead Engineer, AI Compute Infrastructure

ByteDance • San Jose (CA)

On-site
USD 244,800 - 450,000
Day‑one health benefits
401(k) with company match
Parental leave
+1
Senior AI Infra Engineer - Developer Tooling
Senior AI Infra Engineer - Developer Tooling

ByteDance • San Jose (CA)

On-site
USD 212,800 - 387,600
Medical insurance
Dental insurance
Vision insurance
+8
Backend Engineer, AI Systems & Memory Orchestration
Backend Engineer, AI Systems & Memory Orchestration

TwinMind AI, Inc. • Menlo Park (CA), Northern (KY)

Hybrid
USD 150,000 - 230,000
Unlimited PTO
Health Insurance
AI Security Platform Engineer
AI Security Platform Engineer

ByteDance • San Jose (CA)

On-site
USD 140,000 - 230,000