Staff Machine Learning Engineer, Agent Memory & Reasoning

This is a fully remote position, open to applicants in United States.

📋 Description

• Develop agent memory systems: focusing not only on selecting the context but also on the processes that create, manage, refine, and retain that information initially.

• Create memory systems with genuine constraints: ensuring confidentiality and scoping so agents do not disclose sensitive information.

• Establish systems that monitor agent behavior within the organization and translate that into scalable shared best practices.

• Construct reusable "skills" that agents can utilize: including enhanced reasoning, improved financial decision-making, and superior report writing.

• Design and execute tests and benchmarks to validate the effectiveness of these enhancements.

• Contribute to shaping the technical roadmap for agent memory and reasoning as the team develops.

• Transform undefined challenges into concrete, deployable solutions.

• Clearly document your approach so that others can build upon it.

⛳️ Requirements

• You have successfully launched production agents or related systems at a company, beyond experimental settings.

• Proficient in memory, context engineering, or methods that enhance agents' reasoning abilities without necessitating retraining.

• Possess a practical, builder's mindset: focused on rigorous problem-solving with deliverables in days and weeks rather than semesters.

• Capable of independently managing ambiguous, high-level problems.

• Strong foundation in software engineering complemented by your experience in ML and agent development.

• Practical knowledge of the contemporary agent tooling ecosystem: vector databases (Pinecone, Weaviate, pgvector, etc.), retrieval frameworks (LangChain, LlamaIndex), and agent orchestration tools like LangGraph.

• Comfortable interacting directly with LLM provider APIs (OpenAI, Anthropic, etc.) and embedding models for retrieval and memory systems.

• Experience with agent evaluation and benchmarking tools (e.g., LangSmith, Ragas, TruLens, or a custom evaluation harness).

• Excellent communication skills, both written and verbal.

🏝️ Benefits

• Health insurance

• Paid time off

• Flexible work arrangements

• Professional development opportunities