Principal Machine Learning Engineer, Memory & Context

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

📋 Description

• Oversee the architecture of agent memory, detailing what the assistant retains and the processes for writing, retrieving, updating, and retiring memories.

• Spearhead context engineering for production agents by organizing, prioritizing, and condensing context.

• Create and manage agents from start to finish, encompassing prompt and context design through to deployment and operational execution.

• Establish offline and online evaluation frameworks to assess memory and context quality.

• Utilize evaluation findings to inform product and technical strategies.

• Monitor emerging paradigms in agentic approaches, memory architectures, and long-context techniques, transforming them into product innovations.

• Drive intricate, multi-team projects.

• Articulate technical direction to engineers, partner teams, and leadership.

⛳️ Requirements

• A master's degree or higher, or equivalent experience, in Computer Science, Electrical Engineering, or a related discipline.

• More than 5 years of experience in deploying large-scale ML systems, including at least the most recent year focused on building and delivering LLM-based agents in a production environment.

• Practical experience in developing agents from end to end: orchestration, prompt engineering and optimization, as well as context engineering.

• Proven experience in evaluating agentic systems and leveraging those insights for enhancement.

• Demonstrated technical leadership with experience in influencing technical direction across teams and mentoring fellow engineers.

• Exceptional communication skills, including the ability to present to senior leadership.

• Strong programming proficiency in Python.

• Knowledge of agentic AI frameworks such as LangGraph, Agents SDK, or AutoGen.

• Candidates must reside in one of the 50 United States, with limited exceptions.

🏝️ Benefits

• Equity awards determined by factors such as experience, performance, and location.

• Flexibility to work remotely from a location of choice.

• Options for flexible work arrangements to enhance productivity.

• An equal employment opportunity in an inclusive and innovative workplace.

• Accommodations for disability or special needs.