Zillow
Principal Machine Learning Engineer (Memory & Context)
$199K–$335K
Remote-USA · Remote · Principal · Full-time · Posted Oct 22, 2026
- Experience
- 5+ years deploying large-scale ML systems, including at least the most recent year building and shipping LLM-based agents in production.
- Education
- Master’s degree or above, or equivalent experience, in Computer Science, Electrical Engineering, or a related field.
About the job
Leads the design, evaluation, and production deployment of agent memory and context systems for Zillow’s AI assistant.
What you'll do
- Shape memory architecture and production context engineering for agents.
- Build and orchestrate agents end to end and develop offline and online evaluation frameworks.
- Lead cross-functional technical initiatives and communicate technical direction.
What they ask for
- Master’s degree or above, or equivalent experience, in Computer Science, Electrical Engineering, or a related field.
- 5+ years deploying large-scale ML systems, including at least the most recent year building and shipping LLM-based agents in production.
- Hands-on experience building agents end to end, evaluating agentic systems, and improving them based on results.
- Technical leadership across teams, mentoring, strong communication, and strong Python programming skills; familiarity with agentic AI frameworks such as LangGraph, Agents SDK, or AutoGen.
Nice to have
- Experience designing agent or user memory systems, long multi-session dialogue systems, or personalization from conversation history.
- Publications, patents, or open-source contributions related to memory, long-context models, or agents.
Pay
$209,400.00 - $334,600.00 USD annually (base salary); $199,000.00 - $317,800.00 USD annually (base salary)
Tools
Python, LangGraph, Agents SDK, AutoGen
AI focus
Develops agent memory and context engineering, agent orchestration, and evaluation for production AI assistants.
Model-written summary · the original posting is authoritative