Skip to content
Mem0Mem0San Francisco, CA

Senior Research Engineer

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

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 language and/or retrieval and generation 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 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

PythonPyTorchvLLMRAGInformation RetrievalLlm Fine-TuningModel TrainingEvaluation MetricsData PipelinesVector Databases

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
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
Garner Health

Senior Software Engineer

Garner HealthUnited States

Senior Software Engineer leads technical strategy, builds AI-optimized systems processing billions of medical records, and mentors engineers. Requires 4+ years experience, generalist mindset, AI coding tools proficiency, and HIPAA familiarity in AWS/Kubernetes environment.

175k – 215k/yr
Remote4+ YOEML Engineering