Principal Machine Learning Engineer (Memory & Context)

About the teamThe Agentic AI team at Zillow is at the forefront of transforming the real estate industry by helping millions of people use AI assistants to find their next home. Within this organization, our team owns how the assistant remembers and understands each user across a home search that can last for months: agent memory, context engineering, and the evaluation of agents running in production. We are a senior, fast-moving team of machine learning engineers who do the science and own taking it to production.

As a Principal Machine Learning Engineer on the Agentic AI team, you will get to:
Own the architecture of agent memory: what Zillows assistant remembers about each user across conversations, and how memories are written, retrieved, updated, and retired
Lead context engineering for production agents: assembling, prioritizing, and compressing context from conversation history, user memory, tools, and skills so every response is relevant, fast, and cost-efficient
Build and orchestrate agents end to end, from prompt and context design to deployment and operation in production
Define evaluation frameworks for memory and context quality, offline and online, and use them to drive product and technical decisions
Stay on the cutting edge of emerging agentic paradigms, including memory architectures and long-context techniques, and convert them into tangible product innovation
This role has been categorized as a Remote position. “Remote” employees do not have a permanent corporate office workplace and, instead, work from a physical location of their choice, which must be identified to the Company. S. employees may live in any of the 50 United States, with limited exceptions.In addition to a competitive base salary this position is also eligible for equity awards based on factors such as experience, performance and location. Actual amounts will vary depending on experience, performance and location. A masters degree or above, or equivalent experience, in Computer Science, Electrical Engineering, or a related field
~5+ years of experience 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: orchestration, prompt engineering and optimization, and context engineering
~ Strong programming skills in Python and familiarity with agentic AI frameworks such as LangGraph, Agents SDK, or AutoGen

Personalization of AI assistants from conversation history
As the most-visited real estate platform in the U.we help people navigate buying, selling, financing and renting with greater ease and confidence. Whether youre working in tech, sales, operations, or design, you’ll be part of a company reshaping an industry and helping more people make home a reality.
How we work is a key part of that transformation. Through Cloud HQ, our strategic approach to distributed work, most employees have the flexibility to work from wherever they’re most productive—enabling us to move fast, stay connected and deliver on our people promise.
No matter where you sit in the organization, your work will help drive innovation, support our customers, and move the industry—and your career—forward, together.
Zillow Group is an equal opportunity employer committed to fostering an inclusive, innovative environment with the best employees. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. If you have a disability or special need that requires accommodation, please contact your recruiter directly.
Qualified applicants with arrest or conviction records will be considered for employment in accordance with applicable state and local law.
communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation.