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
$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...SuggestedFull time$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...SuggestedFull time- ...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
- ...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
$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...SuggestedWork experience placementCurrently hiringWork at officeLocal areaRemote workRelocation packageFlexible hours$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$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 officeLocal areaRelocationRelocation package- ...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...
$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- 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
$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...- ...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
- 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...
$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 timeTemporary workLocal areaWorldwide- 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...
$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 workWork at officeLocal areaRelocationRelocation package$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 officeLocal areaRelocationRelocation package- ...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
$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 hours$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 timeLive inWork at officeLocal area$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 areaWorldwide- ...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 timeWork at officeRemote workFlexible hours
$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 week- 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...
- ...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...
- ...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...InternshipShift work
$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,...- 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...
$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- ...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...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff ML Engineer — Memory & Agent Retrieval. Be the first to apply!
- staff engineer San Francisco, CA
- assistant engineer San Francisco, CA
- research assistant engineering San Francisco, CA
- staff design engineer San Francisco, CA
- staff security engineer San Francisco, CA
- engineering aide San Francisco, CA
- senior staff engineer San Francisco, CA
- senior staff systems engineer San Francisco, CA
- assistant chief engineer San Francisco, CA
- assistant electrical engineer San Francisco, CA

