Experience Senior · 6+ yrs (6+ years)
About the role
structured by ORIZep AIAgent Context Is Hard. We Fixed It.Member of Technical Staff: Research$180K - $250K•1.00% - 1.50%•San Francisco, CA, US / Remote (US)Job typeFull-timeRoleEngineering, Machine learningExperience6+ yearsVisaUS citizen/visa onlySkillsAmazon Web Services (AWS), Python, Torch/PyTorch, Machine Learning, LLMs, AI…
What you will do
- Build agents that improve retrieval: query understanding, ranking, and what to pull into context.
- Finetune the models that extract, update, and consolidate memory on Zep's domain.
- Finetune the models that power those agents, so the retrieval loop and the agent get better together.
- Own the work from dataset creation through experiment design, evaluation, training, and a change that ships to production. Our engineers work the serving path alongside you.
- Build the eval harnesses that catch regressions in retrieval, memory quality, and agent task completion before a release ships.
What they are looking for
- You have shipped production ranking, retrieval, or query understanding.
- You've built classical ML into a real product — logistic regression, SVMs, GBTs, single-layer perceptrons.
- You've shipped finetuned models to production.
- You know transformer architectures and training workflows, and you work in PyTorch.
- You have shipped a non-trivial agentic system to production. Not a prototype, not a thin wrapper over a chat-completion API.
- You have a research methodology: dataset creation and curation, experiment design, and evaluation. You can frame an open problem and design an experiment that answers it.
- You have built evaluation for retrieval, generation, or agent tasks: gold sets, offline metrics, online tests.
- You write strong Python, and you have enough production AWS to ship, monitor, and iterate on what you train.
- You have a Master's in Computer Science or equivalent experience.
Amazon Web Services (AWS)PythonTorch/PyTorchMachine LearningLLMsAI Agentslogistic regressionSVMsAWSPyTorch
Full posting text
Experience: 6+ years
Zep manages, governs, and serves agent memory at enterprise scale. We're hiring an engineer who builds agents and trains the models they run on.