Search Remote Jobs

Principal Machine Learning Engineer, Memory & Context

🔥 13 hours ago

🇺🇸 United States – Remote

đź’µ $199k - $334.6k / year

⏰ Full Time

đź”´ Lead

🤖 Machine Learning Engineer

Apply Now
Find Similar Remote Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Logo of Zillow

Zillow

5001 - 10000 employees

Founded 2006

🏠 Real Estate

🛍️ eCommerce

👥 B2C

đź’° $4.1M Post-IPO Equity on 2012-12

Real Estate • eCommerce • B2C

Zillow is a leading real estate and property rental marketplace that provides comprehensive information on homes, apartments, and properties for sale or rent. It offers users tools to search for properties, calculate mortgage rates, and connect with real estate agents. The platform also features innovative algorithms that provide Zestimates, which are estimated market values of homes. Zillow is a go-to resource for individuals looking to buy, sell, or rent properties, as well as for agents and brokers who want to reach a wider audience.

đź“‹ Description

• Own the architecture of agent memory, including what the assistant remembers and how memories are written, retrieved, updated, and retired • Lead context engineering for production agents by assembling, prioritizing, and compressing context • Build and orchestrate agents end to end, from prompt and context design to deployment and production operation • Define offline and online evaluation frameworks for memory and context quality • Use evaluation results to drive product and technical decisions • Track emerging agentic paradigms, memory architectures, and long-context techniques and convert them into product innovation • Lead complex, multi-team initiatives • Communicate technical direction to engineers, partner teams, and leadership

🎯 Requirements

• 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 and experience influencing technical direction across teams and mentoring other engineers • Strong communication skills, including presenting to senior leadership • Strong programming skills in Python • Familiarity with agentic AI frameworks such as LangGraph, Agents SDK, or AutoGen • Employees must reside in one of the 50 United States, with limited exceptions

🏖️ Benefits

• Equity awards based on factors such as experience, performance and location • Remote work from a physical location of choice • Flexible work from wherever employees are most productive • Equal employment opportunity and inclusive, innovative environment • Disability or special-needs accommodation

Apply Now

Similar Jobs

đź•’ Yesterday

Spotify

5001 - 10000

📱 Media

👥 B2C

🛍️ eCommerce

Staff Machine Learning Engineer building Spotify podcast recommendations across Home and Now Playing. Developing large language models and production-scale personalization systems for millions of listeners.

đź•’ 6 days ago

Cushman & Wakefield

10,000+ employees

đź’Ľ Consulting

🏥 Healthcare

🏗️ Construction

Principal ML Engineer building production AI, agentic systems, and MLOps on Databricks and Azure. Advancing governed machine learning for Cushman & Wakefield’s real estate services.

đź•’ October 12

DISCO

501 - 1000

🤖 Artificial Intelligence

🤝 B2B

⚖️ Legal

Principal Machine Learning Scientist leading machine learning research and AI strategy for DISCO, a cloud-native legal technology company. Driving algorithms, research programs, and solutions for ediscovery and legal case management.

đź•’ October 9

Natera

1001 - 5000

🏥 Healthcare

🧬 Biotechnology

⚕️ Healthcare Insurance

Staff scientist developing machine learning and bioinformatics models for Natera’s cfDNA cancer detection tests. Advancing genomic research from prototypes through clinical validation.

đź•’ October 2

Reddit, Inc.

501 - 1000

đź’Ľ Consulting

📣 Marketing

📱 Media

Staff ML Engineer building GenAI and predictive ad-creative products for Reddit. Developing image, video, and copy generation systems that improve advertiser campaign performance.