Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff ML Engineer — Memory & Agent Retrieval

Zep AI

Zep AI in San Francisco seeks an engineer who builds agents and trains the models they run on. You will own the memory extraction and retrieval loop, finetune models, and push systems to production. You’ll report to the founder and join a team with experience at Scale AI, Dropbox, and McKinsey. You will work in a distributed team, write production-grade Python, design experiments, and build eval harnesses to catch regressions before release. A Master’s in CS or equivalent is required. #J-18808-Ljbffr Zep AI

Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Staff ML Engineer — Memory & Agent Retrieval in San Francisco, CA vacancy
  • $264.8k - $331k

     ...capabilities.About the General Agents TeamThe General Agents team...  ....About the RoleAs a Senior/Staff Machine Learning Engineer (MLE) on the General Agents...  ...LLM reasoning, tool use, memory, and control logic to solve...  ....Experience deploying ML systems in cloud environments... 
    Suggested
    Full time

    Scale AI

    San Francisco, CA
    10 hours ago
  • $252k - $315k

     ...build production AI agents that automate complex...  ...one of the hardest engineering challenges.As a Staff Frontier Agent Engineer...  ...knowledge, retrieval, traditional machine...  ....Unlike traditional ML roles that focus on...  ...reasoning, retrieval, memory, and tool use.Architect... 
    Suggested
    Full time

    Scale AI

    San Francisco, CA
    2 days ago
  •  ...pipeline that identifies the best web query results. The ideal candidate will possess a strong intuition for relevance and feature engineering. This position requires a rigorous approach to evaluation based on real-world outcomes. At Parallel, the work is fully in-person,... 
    Suggested

    Parallel

    San Francisco, CA
    10 hours ago
  •  ...AI models and a high-efficiency serving platform. We seek an ML Engineer who operates at the intersection of production software...  ...systems. You will design architectures that integrate LLMs, retrieval, memory, tools, and feedback loops, and contribute to open-source projects... 
    Suggested

    Sciforium

    San Francisco, CA
    2 days ago
  • $189.31k - $389.75k

     ...new possibilities and plan for memories that will last a lifetime. At Pinterest...  ...focuses on applied ML research and development. Labs...  ...in-house VLM and composed image retrieval models. The core visual pod is a small group (~10 engineers) inside Labs, which allows for... 
    Suggested
    Work experience placement
    Currently hiring
    Work at office
    Local area
    Remote work
    Relocation package
    Flexible hours

    Pinterest

    San Francisco, CA
    1 day ago
  • $197.3k - $313.7k

     ...AI CRM, where humans with agents drive customer success together...  ...*Slack is looking for a Staff Machine Learning Engineer with deep expertise in model...  ...and finetuning to join our ML team. You'll design, train,...  .../ONNX).Experience with retrieval-augmented generation and hybrid... 
    Full time

    Salesforce

    San Francisco, CA
    4 days ago
  • $189.31k - $389.75k

     ...possibilities and plan for memories that will last a...  ..., and machine learning engineers. You'll conduct research...  ...problems, especially search agent, recommendation agent,...  ...in Computer Science, ML, NLP, Statistics, Information...  ...learning/information retrieval Mastery of at least one... 
    Work at office
    Local area
    Relocation
    Relocation package

    Pinterest

    San Francisco, CA
    4 days ago
  •  ...product platform for enterprise clients. You will design and implement the core ML reasoning, retrieval, orchestration, evaluation, and fine-tuning components that power our intelligence engine. You will ship research-backed techniques to production, own experiments and... 

    Crucibl

    San Francisco, CA
    4 days ago
  • $264.8k - $331k

    Staff Machine Learning Research Engineer, Agent Post-training - Enterprise GenAI AI is becoming vitally important in every function of our society. At Scale...  ...agents in enterprises around the world. The Enterprise ML Research Lab works on the front lines of this AI... 
    Full time

    Scale

    San Francisco, CA
    3 days ago
  • United States Digital Space LLC in San Francisco is seeking a Staff ML Engineer (IC6) to define and lead the foundation model strategy...  ..., mentor engineers, and drive cross-team initiatives spanning retrieval, ranking, #J-18808-Ljbffr United States Digital Space LLC

    United States Digital Space LLC

    San Francisco, CA
    4 days ago
  • $190k - $265k

     ...San Francisco is looking for a Research Engineer to enhance their AI platform for law, tax...  ...involve developing low-latency search and retrieval systems and managing large datasets of complex...  ...candidate will have over 5 years of ML engineering experience and proven ability... 

    Harvey

    San Francisco, CA
    10 hours ago
  •  ...operating system that heals, evolves, and documents codebases autonomously. The Founding ML Engineer will architect intelligence powering autonomous pull requests, long-term memory, and self-improving workflows across product, infra, and full-stack teams. This in-person... 
    Remote work

    Kodezi Inc.

    San Francisco, CA
    3 days ago
  • Glean is hiring a Machine Learning Engineer to advance the AI Assistant and autonomous agents. You will operate at the crossroads of production ML, LLM-powered systems, and product engineering to build, evaluate, and iterate on assistant experiences grounded in real enterprise... 

    Gleanwork

    San Francisco, CA
    2 days ago
  • $190.2k - $345.65k

     ...and, increasingly, for AI agents — is becoming one of the product...  ....We are hiring a Senior Staff Machine Learning Engineer to architect and lead the...  ...expose it as agentic search: retrieval designed to be driven by AI...  ...loops that improve them.ML Engineering leadership & AI... 
    Full time
    Temporary work
    Local area
    Worldwide

    Adobe Systems

    San Francisco, CA
    5 hours ago
  • Perplexity is seeking experienced ML engineers to design, build, and optimize the recommendation...  ...integrating LLMs into ranking, retrieval, or personalization pipelines. Taste and...  ...something and hope. They must have armies of agents and workers who can constantly work in millions... 

    Perplexity AI Inc.

    San Francisco, CA
    1 day ago
  • $227.87k

     ...find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we’re on a mission to...  ...experience.7+ years of industry experience.Strong software engineering and mathematical skills with knowledge of statistical methods.... 
    Temporary work
    Work at office
    Local area
    Relocation
    Relocation package

    Pinterest

    San Francisco, CA
    3 days ago
  • $268.08k

     ...about new possibilities and plan for memories that will last a lifetime. At Pinterest...  ...here.We are looking for a Sr. Staff Machine Learning Engineer to be the Technical Lead for the Content...  ...making workflows.Partner closely with ML engineers to translate ideas into production... 
    Work at office
    Local area
    Relocation
    Relocation package

    Pinterest

    San Francisco, CA
    3 days ago
  •  ...that runtime. As a Senior Machine Learning Engineer for On-Device & Mobile AI, you will take...  ...interactive frame rates within a fixed memory and power budget. Your work directly shapes...  ...parts of the integration between the ML runtime and the game engine: real-time scheduling... 
    Full time
    Work at office
    Remote work
    Worldwide

    Unity Technologies

    San Francisco, CA
    1 day ago
  • $242.8k - $357k

     ...and discovery experiences. As a Senior Staff Engineer, you will lead the technical direction for...  ...understanding.Rigorously evaluate ML and LLM models using a combination of offline...  ...learning, large language models, information retrieval, ranking and relevance, recommendation... 
    Hourly pay
    Work at office
    Local area
    Remote work
    Flexible hours

    Doordash

    San Francisco, CA
    1 day ago
  • $229k - $343k

     ...other digital services.Snap Engineering teams build fun and technically...  ....​We’re looking for a Staff Machine Learning Engineer to...  ...such as learning-to-rank, deep retrieval, neural ranking, sequence models...  ...mentor engineers working on ML ranking systemsStay current with... 
    Full time
    Live in
    Work at office
    Local area

    Snap

    San Francisco, CA
    4 days ago
  • $190k - $285k

     ...Databricks Assistant, AI/BI Genie, and Agent Bricks working with product...  ...we are seeking multiple GenAI Engineers from junior levels to more...  ...domains.Design and implement ML pipelines for data...  ...tuning, prompt engineering, and retrieval-augmented generation (RAG) is... 
    Work at office
    Local area
    Worldwide

    DataBricks

    San Francisco, CA
    10 hours ago
  •  ...educational institutions Work alongside engineers, scientists, operators, and...  ...Role Handshake is hiring a Staff Machine Learning Engineer for...  ...multiplier - architecting the ML infrastructure that the...  ...of systems built on multiple retrieval models - Graph Neural Network,... 
    Full time
    Work at office
    Remote work
    Flexible hours

    AI Chopping Block

    San Francisco, CA
    1 day ago
  • $227.33k - $312.58k

    We’re looking for a Staff ML Data Engineer to join Procore’s AI & Frontier Models organization. In this role, you’ll be responsible for designing and building the data systems that power frontier‑scale machine learning research and applied AI products, with a particular... 
    Full time
    Work at office
    Local area
    Immediate start
    3 days per week

    Procore Technologies

    San Francisco, CA
    2 days ago
  • Reddit, Inc. is seeking a Staff Machine Learning Engineer for Shopping Ads to lead the technical strategy and execution of ML models powering Shopping Ads delivery. You will own end...  ...to deployment, spanning targeting, retrieval, ranking, and value optimization across... 

    Reddit

    San Francisco, CA
    10 hours ago
  •  ...Liquid team is a community of world-class engineers, researchers, and builders creating the...  ...Ability to write and debug models in popular ML frameworks, and experience working with...  ...: on CPUs, with low latency, minimal memory, and maximum reliability. We’re already partnering... 

    Liquid AI

    San Francisco, CA
    10 hours ago
  •  ...hardware, ensuring low latency, minimal memory usage, privacy, and reliability. We partner...  ...engagement, working directly with the engineer who leads our embedded customer R&D. You...  ...automotive, embedded-device, or on-device ML context, and the instinct to reason from... 
    Internship
    Shift work

    Doist

    San Francisco, CA
    2 days ago
  • $268.08k

    Pinterest is seeking a Sr. Staff Machine Learning Engineer to serve as the Technical Lead for Content Quality. This role involves building the technical...  ...at scale. The successful candidate will partner with ML engineers, develop a roadmap for signal quality and trust,... 

    Pinterest

    San Francisco, CA
    10 hours ago
  • HausHaus is seeking a senior ML Engineer to drive high-impact projects in advanced marketing planning, analysis, and optimization. You will collaborate with applied scientists, data scientists, and engineers to deliver trustworthy results, building scalable ML systems for... 

    HAUS

    San Francisco, CA
    4 days ago
  • $220k - $280k

     ...powers production-grade, real-time voice agents and applications — serving speech-to-...  ...latency and reliability. We're looking for a Staff ML Engineer to drive the model serving layer for...  ..., streaming inference pipelines, and memory management tailored to real-time audio;... 
    Full time

    Together

    San Francisco, CA
    2 days ago
  •  ...building reinforcement learning-based LLM training pipelines. You will train models to tackle challenges in materials science and engineering. Qualifications include industry experience with Python and PyTorch, strong communication skills, and an engineering mindset. The... 

    Outerport

    San Francisco, CA
    3 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff ML Engineer — Memory & Agent Retrieval. Be the first to apply!