Skip to content
Mem0Mem0San Francisco, CA

Senior Research Engineer

Owns end-to-end lifecycle of memory features in AI systems, fine-tuning LLMs for extraction/updates/forgetting, implementing research papers, building large-scale evaluations, and shipping production models with engineering. Requires RAG/IR experience, PyTorch proficiency, and production ML expertise.

175k – 210k/yr
On-siteML Engineering

About the role

What You'll Do

  • Fine-tune and train models for memory extraction, updates, consolidation/forgetting, and conflict resolution; iterate based on data and outcomes.
  • Read, reproduce, and implement research: quickly prototype paper ideas, benchmark against baselines, and productionize what wins.
  • Build evaluation at scale: automated relevance/accuracy/consistency metrics, gold sets, online A/B & interleaving, and clear dashboards.
  • Work closely with customers to uncover pain points, turn them into research hypotheses, and validate solutions through field trials.
  • Partner with Engineering to ship: design APIs and data contracts, plan safe rollouts, and maintain SOTA latency, reliability, and cost at scale.

Minimum Qualifications

  • Experience in RAG or information retrieval (retrieval, ranking, query understanding) for real products.
  • Model training/fine-tuning experience (LLMs/encoders) with a strong footing in experimental design and iteration.
  • Strong Python; deep experience with PyTorch and familiarity with vLLM and modern serving frameworks.
  • Built evaluation for complex vision-and-language tasks (gold sets, offline metrics, online tests).
  • Able to orchestrate data pipelines to run these models in production with low-latency SLAs (batch + streaming).
  • Clear, concise communication with stakeholders (engineering, product, GTM, and customers).

Nice to Have

  • Publications at venues like CVPR, NeurIPS, ICML, ACL, etc.
  • Experience with privacy-preserving ML (redaction, differential privacy, data governance).
  • Deep familiarity with memory/retrieval literature or prior work on memory systems.
  • Expertise with embeddings, vector-DB internals, deduplication, and contradiction detection.

Skills

RAGPyTorchPythonLLMsvLLMInformation RetrievalVector DatabasesEmbeddingsFine-TuningA/B TestingData Pipelines

Similar roles

ML Engineering jobs
Zilliz

Senior Software Engineer, Vector Index Research

ZillizRedwood City, CA

Research and implement high-performance vector indexing and retrieval algorithms for Milvus and Zilliz Cloud. Requires 3+ years in vector search or HPC, strong C++ or Rust skills, and a research-driven engineering mindset.

175k – 250k/yr
Hybrid3+ YOEML Engineering
Mem0

Senior Research Engineer

Mem0San Francisco, CA

Own the end-to-end lifecycle of memory features for AI agents. Fine-tune models, implement research, build evaluations, and ship production systems with Engineering.

175k – 250k/yr
On-site7+ YOEML Engineering
AKASA

Sr. Machine Learning Engineer

AKASASouth San Francisco, CA

As a Senior Machine Learning Engineer, you will develop and deploy state-of-the-art ML solutions for healthcare problems, working with large medical datasets and owning ML services end-to-end. This role requires expertise in LLMs, cloud platforms, and ML frameworks.

175k – 230k/yr
Hybrid5+ YOEML Engineering
Applied Intuition

Applied Perception Engineering Lead

Applied IntuitionAnn Arbor, MI

Lead a team of software engineers to develop and integrate perception pipelines for various modalities and an autonomy service for government and defense applications. This role focuses on deploying software to hardware and integrating pre-trained ML models.

175k – 210k/yr
On-site8+ YOEML Engineering
Applied Intuition

Applied Perception Engineering Lead

Applied IntuitionSan Diego, CA

Lead a team of software engineers to develop and integrate perception pipelines for various modalities and an autonomy service for government and defense applications. This role focuses on deploying software to hardware components and integrating pre-trained ML models.

175k – 235k/yr
On-site8+ YOEML Engineering