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.
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
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.
175k – 235k/yr
On-site8+ YOEML Engineering
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.