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
- ENGRAM Lab is seeking an experienced ML Engineer to build and deploy memory retrieval models and scalable ML infrastructure. You will design and implement models, train pipelines, and deploy them to production at scale, ensuring real-time performance and robust monitoring...Suggested
- ...Staff ML Engineer, Agent Training & Environments Shape the Future of AI At Labelbox, we're building the critical infrastructure that powers breakthrough AI models at leading research labs and enterprises. Since 2018, we've been pioneering data-centric approaches...SuggestedFlexible hours
WeHireYou
San Francisco, CA1 day ago $290.4k - $363k
...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...SuggestedFull timeScale AI
San Francisco, CA6 days ago$265k - $331k
...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...SuggestedFull timeScale AI
San Francisco, CA5 days ago- Labelbox is the RL data factory for advancing frontier agent capabilities. We build the data, environments, and evaluations that frontier labs use to train and judge their agents. This Staff ML Engineer role sits where training meets infrastructure, running experiments...Suggested
- ...environments, evaluators, and pipelines that enable scalable experimentation and rapid iteration. This role blends platform engineering with agent-focused post-training expertise, requiring engineers who can ship reliable systems at scale and drive architectural...
- ...property management. We build AI agents that act as property... ...and most valuable kind of AI engineering there is. What You'll Do... ...engineering, tool/function calling, memory systems, retrieval‑augmented generation (RAG),... ...AI systems. Classical ML experience — supervised/...Live inRemote work
$189.31k - $389.75k
...possibilities and plan for memories that will last a lifetime.... ...looking for an experienced Staff Machine Learning Engineer / Technical Lead who can drive... ...leader who combines strong ML judgment with product... ...self-supervised learning, retrieval, and distillation.Partner with...Work at officeLocal areaRelocation packagePinterest
San Francisco, CA3 days ago$222.72k - $389.75k
...possibilities and plan for memories that will last a lifetime.... ...process here.We’re hiring a Staff Machine Learning Engineer to help drive the future of... ...and focuses on building AI/ML systems (including LLMs) that... ...clear downstream impact on retrieval, ranking, and shopping...Work at officeLocal areaRelocationRelocation packagePinterest
San Francisco, CA6 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...
$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...- Anthropic is seeking a Software Engineer on the Account Abuse team to build ML systems for detecting and preventing account abuse at scale. You will develop... ..., ensuring point-in-time correctness and low-latency retrieval. You will train and deploy models offline and online...
- ...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 timeWork at officeRemote workWorldwide
Unity Technologies
San Francisco, CA7 days 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 payWork at officeLocal areaRemote workFlexible hoursDoordash
San Francisco, CA2 days ago$222.72k - $389.75k
...dream about new possibilities and plan for memories that will last a lifetime. At Pinterest,... ...for Pinterest ads. We’re looking for an ML engineer with experience in ads delivery, machine... .../ utility, or machine learning.Proven Staff-level technical leadership as an IC:...Work at officeLocal areaRelocationRelocation packagePinterest
San Francisco, CA4 days ago$227.87k
...about new possibilities and plan for memories that will last a lifetime. At Pinterest... ...PinFlex landing page to learn more. Staff Machine Learning Engineer, Applied Science As part of Pinterest... ...’ll do:Contribute to cutting-edge AI/ML research that can be applied to...Work at officeLocal areaRemote workRelocationRelocation packageFlexible hoursPinterest
San Francisco, CA3 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 officeLocal areaWorldwideDataBricks
San Francisco, CA6 days ago$172.5k - $306.63k
Join Adobe Security Engineering and help build the ML and generative AI capabilities that... ...'s largest datasets.As a Staff Machine Learning Engineer,... ...investigation, including retrieval-augmented generation.... ...databases, RAG, or multi-agent systems.Publications, patents...Full timeTemporary workLocal areaWorldwideAdobe Systems
San Francisco, CA3 days ago- ...AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - Pre-training Infrastructure San Francisco Bay Area, California, United States... ...Profile and optimize JAX/XLA workloads across compute, memory, communication, and compilation. Develop and optimize...
Apple Inc.
San Francisco, CA1 day 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...Work at officeRemote workWorldwideRelocation package
Unity Enterprise
San Francisco, CA1 day ago $205k - $240k
...We're looking for a Senior Staff Machine Learning Engineer to lead the AI behind Altana... ...build the models and AI agents that turn it into real-time... ...shape how we hire and level ML talent. Shape the product... ...production LLM applications: retrieval-augmented generation (RAG),...Full timeTemporary workWork experience placementRemote workFlexible hoursAltana
San Francisco, CA2 days ago$210k - $260k
...their financial journey. Machine Learning Engineers at Rocket Money further our mission by... ...teams, know how to support strategy with ML and AI powered user experiences, can... ...effects of their products on end users. At the Staff level, Machine Learning Engineers are...Work at officeLocal areaRocket Money
San Francisco, CA1 day ago$300k
...casual encounter or long-term relationship. We’re doubling down on ML as the future of Grindr, and in these early days, your work... ...problems for our user. Collaborate cross-functionally with engineering, data science and product teams to turn bold ideas into tangible...Casual workWork at officeImmediate startWorldwideFlexible hoursGrindr
San Francisco, CA4 days ago- ...data. At Plenful, we build agents and models that do that... ...wherever judgment matters. As a Staff Machine Learning Engineer, you set the technical... ...production agents: tool use, retrieval, state and context... ...everyone Co‑own the 12‑month ML roadmap with engineering and...Full timeWork at officeRemote workFlexible hours2 days per week
Plenful
San Francisco, CA1 day ago - ...Staff Machine Learning Engineer Primer exists to make the world a safer place. We... ...language models, agentic systems, retrieval, embeddings, fine-tuned... ...what frontier models and agents make newly possible. You reach... ...of shipping and operating ML-driven functionality. ~...Contract workRemote workFlexible hours
Primer
San Francisco, CA5 days ago $273k - $345k
...improve them until they work at scale. We are roboticists, engineers, operators, and builders. We believe the next great technology... ...time inference pipelines to identify and eliminate CPU, GPU, and memory bandwidth bottlenecks on the vehicle. Work with low-level hardware...Full timeInternshipWork at officeFlexible hoursATOMS Careers page
San Francisco, CA1 day ago$232k
...Rides, Eats, Grocery, and more. As a Staff Machine Learning Engineer, you will shape the technical and... ...and lead the design and development of ML solutions, such as real-time... ...solutions that go beyond traditional retrieval and ranking approaches. You will partner...Full timeWork at officeRemote workUber Technologies Inc.
San Francisco, CA1 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 timeWork at officeLocal areaImmediate start3 days per weekProcore Technologies
San Francisco, CA3 days ago- A leading AI infrastructure company in San Francisco is looking for a Machine Learning Engineer with expertise in model development and scalable ML systems. The role involves creating essential components for multimodal data platforms and enhancing the developer experience...Full time
$212k - $276.5k
...and adaptive user experiences. As a Staff Machine Learning Engineer, you will lead complex AI initiatives... ...experience, including proven leadership on AI/ML projects at scale. Hands-on... ...generative AI solutions, emphasizing Retrieval-Augmented Generation (RAG), agentic workflows...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff ML Engineer — Memory & Agent Retrieval. Be the first to apply!
Related searches
- project engineer assistant project manager San Francisco, CA
- staff security engineer San Francisco, CA
- staff engineer San Francisco, CA
- research assistant engineering San Francisco, CA
- senior staff systems engineer San Francisco, CA
- engineering aide San Francisco, CA
- senior staff engineer San Francisco, CA
- assistant engineering manager San Francisco, CA
- assistant chief engineer San Francisco, CA
- staff data engineer San Francisco, CA