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.
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
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
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 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 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.