Staff ML Engineer - Agent Memory & Reasoning Systems

Wand AI; Palo Alto (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

Wand AI in Palo Alto seeks a Staff Machine Learning Engineer to advance agent memory and reasoning capabilities. You will drive memory architecture, context engineering, and scalable infrastructure for AI agents operating alongside humans.

Collaboration with a world-class team will shape practical, high-impact solutions. Responsibilities include building memory systems, enforcing confidentiality, creating reusable agent skills, and benchmarking improvements while guiding the team’s technical

Qualifications

• Shipped production agents or agent-adjacent systems at a company, not just in a lab.
• Experience with memory, context engineering, or techniques that make agents reason better without retraining them.
• An applied, builder's mindset: rigorous thinking, shipped in days and weeks, not semesters.
• Comfortable owning ambiguous, senior level problems on your own.
• Strong software engineering fundamentals to go with your ML and agent experience.
• Practical fluency with the modern agent tooling stack: vector databases, retrieval frameworks, and agent orchestration tools.

Responsibilities

• Build agent memory systems: mechanisms that generate, curate, refine, and store information.
• Design memory with confidentiality and scoping to prevent leaks.
• Build systems that monitor agent behavior across the org and share best practices at scale.
• Create reusable skills agents can call on: better reasoning, decisions, reporting.
• Design and run tests and benchmarks demonstrating improvements.
• Help shape the technical roadmap for memory and reasoning as the team scales.
• Take undefined problems and craft shippable solutions.
• Document methodology clearly for future teams to build on.

Skills

Memory engineering

Context engineering

Agent reasoning

Software engineering

LLM APIs

Vector databases

LangChain

OpenAI API

Education

MS or PhD

Tools

Pinecone

Weaviate

pgvector

LangGraph

LangChain

Job description

Wand AI in Palo Alto seeks a Staff Machine Learning Engineer to advance agent memory and reasoning capabilities. You will drive memory architecture, context engineering, and scalable infrastructure for AI agents operating alongside humans.

Collaboration with a world-class team will shape practical, high-impact solutions. Responsibilities include building memory systems, enforcing confidentiality, creating reusable agent skills, and benchmarking improvements while guiding the team’s technical