Jobs › Companies › Zillow › Principal Machine Learning Engineer (Memory & Context)

About this Principal Machine Learning Engineer (Memory & Context) role at Zillow

Zillow · Remote · Remote-USA

About the team

The 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. We aim to redefine the home shopping and transaction experience by providing an always-on assistant that integrates personalized user insights with Zillow's knowledge. 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.

About the role

As a Principal Machine Learning Engineer on the Agentic AI team, you will get to:


  • Own the architecture of agent memory: what Zillow's 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
  • Lead complex, multi-team initiatives and communicate technical direction clearly to engineers, partner teams, and leadership

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. U.S. employees may live in any of the 50 United States, with limited exceptions.

In California, Connecticut, Maryland, Massachusetts, New Jersey, New York, Washington state, and Washington DC the standard base pay range for this role is $209,400.00 - $334,600.00 annually. This base pay range is specific to these locations and may not be applicable to other locations. In Colorado, Hawaii, Illinois, Maine, Minnesota, Nevada, Ohio, Rhode Island, Vermont, and Virginia the standard base pay range for this role is $199,000.00 - $317,800.00 annually. The base pay range is specific to these locations and may not be applicable to other locations.

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. Employees in this role will not be paid below the salary threshold for exempt employees in the state where they reside.

Who you are

You are a hands-on technical leader with high agency: when a problem is ambiguous or blocked, you go talk to the people who can solve it, and you make your work visible.


Must-haves

  • A master's 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
  • Experience evaluating agentic systems and using those results to improve them
  • Proven technical leadership: you have influenced technical direction across teams and mentored other engineers
  • Strong communication skills, including presenting to senior leadership
  • Strong programming skills in Python and familiarity with agentic AI frameworks such as LangGraph, Agents SDK, or AutoGen

Nice-to-haves

  • Experience designing agent or user memory systems (long-term, episodic, or user-profile memory)
  • Experience with dialogue systems that sustain long, multi-session conversations
  • Personalization of AI assistants from conversation history
  • Publications, patents, or open-source contributions related to memory, long-context models, or agents

Get to know us

At Zillow, we’re reimagining how people move—through the real estate market and through their careers. As the most-visited real estate platform in the U.S., we help people navigate buying, selling, financing and renting with greater ease and confidence. Whether you're 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. 


Zillow is honored to be recognized among the best workplaces in the U.S. Zillow was named one of FORTUNE 100 Best Companies to Work For® in 2026, and included on TIME’s America’s Best Companies 2026 list, reflecting our commitment to creating an innovative, inclusive, and engaging culture where employees are empowered to grow.


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.


Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; 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. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Ready to apply to Zillow?
Apply to Zillow

How this ML Engineer salary compares

This role pays $258,400/yr — above the typical range for ML Engineer roles.

$126,378 median $200,000 $289,665

Typical range $158,625–$244,563/yr, from 1,039 comparable ML Engineer listings on JobsRadar (pay annualized to USD). See ML Engineer salary insights →

About Zillow

At Zillow, flexibility isn’t a perk - it’s how we work. Cloud HQ is our distributed work model, built on trust, clear systems, and the belief that you can do great work from wherever you are. It’s not about where you work. It’s about moving forward - together.

See all jobs at Zillow →

Similar jobs

Airbnb
Machine Learning Engineer, Assistance Engineering
Airbnb
⚡ Apply early Remote-USA · location restricted $162,000–$186,000
● New 👁 Seen ✓ Applied 17h ago
Airbnb
Senior Machine Learning Engineer, Payments
Airbnb
⚡ Apply early Remote-USA · location restricted $191,000–$223,000
● New 👁 Seen ✓ Applied 2w ago
Zillow
Senior Machine Learning Engineer, Rich Media Experiences
Zillow
⚡ Apply early Remote-USA · location restricted $163,200–$260,600
● New 👁 Seen ✓ Applied 1mo ago
Zillow
Principal Machine Learning Engineer, Agentic AI
Zillow
⚡ Apply early Remote-USA · location restricted $194,200–$310,200
● New 👁 Seen ✓ Applied 3mos ago
Pinely
Senior Research Engineer (ML Infrastructure)
Pinely
⚡ Apply early Amsterdam, North Holland, Neth... Onsite
● New 👁 Seen ✓ Applied 2h ago
Ema
Software Engineer, Machine Learning
Ema
⚡ Apply early Mountain View (Hybrid) Hybrid $160,000–$220,000
● New 👁 Seen ✓ Applied 3h ago
Snap Inc.
Software Engineer, ML infrastructure, Core Inference, Level 4
Snap Inc.
⚡ Apply early Palo Alto - 395 Page Mill Rd Onsite $157,000–$235,000
● New 👁 Seen ✓ Applied 3h ago
Faculty
Lead Machine Learning Engineer (Safety)
Faculty
⚡ Apply early London Hybrid
● New 👁 Seen ✓ Applied 5h ago
Redwood Materials
Software Engineer - ML/Computer Vision (Battery Sorting)
Redwood Materials
⚡ Apply early McCarran, NV; San Francisco, C... Onsite $152,500–$200,000
● New 👁 Seen ✓ Applied 5h ago

Sign up for suggestions tailored to the jobs you open and the searches you save.

More jobs at Zillow

See all jobs at Zillow →

Apply now
🤖

Whoa — hold up

JobsRadar was built for real people having a rough time in their job search — not for automated requests. You're clicking way too fast and you're now temporarily blocked.

Come back later. If you're genuinely job hunting, we've got your back — just act like a human.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Get an edge on your job hunt.

Join our Telegram channel for the stuff that helps you land the role — salary benchmarks, the weekly market pulse, and new-feature drops. No spam, just signal.

Join the channel — it's free