Staff AI Engineer - Agent Architecture & Behavior
Staff AI Engineer responsible for designing and shipping agent architecture, multi-agent coordination, reliable execution, memory, tooling, and evaluation systems. Requires substantial shipped agentic-system experience and strong software engineering skills in Python, TypeScript, or a comparable language.
About the job
Responsibilities
- Design and implement agent execution loops, planning strategies, tool interfaces, verification, and system boundaries.
- Build multi-agent delegation, coordination, context sharing, result synthesis, concurrency handling, cancellation, and stale-result management.
- Develop resilient stateful workflows with checkpoints, recovery strategies, partial-failure handling, and human intervention.
- Improve retrieval, context construction, persistent state, memory, and reusable skill representation.
- Build evaluations, analyze task trajectories, and measure quality, reliability, latency, and cost improvements.
- Evaluate models and emerging techniques; make build-versus-buy decisions and replace tools when evidence changes.
- Set engineering standards, review architecture and design decisions, and provide technical leadership as an individual contributor.
- Partner with product and infrastructure engineers on production services, integrations, secure execution, and observability.
Requirements
- Demonstrated experience building and shipping a substantial agentic system through production use or rigorous, reproducible open-source work.
- Deep practical experience with LLM tool use, planning, context engineering, evaluations, and multi-agent coordination or parallel agent/tool execution.
- Hands-on experience with browser or computer automation in an agentic system, including state observation, effect verification, and failure recovery.
- Strong software engineering skills in Python, TypeScript, or a comparable language.
- Experience with asynchronous services, state machines, persistence, concurrency, retries, and cancellation.
- Ability to distinguish model decisions from guarantees that must be enforced in code, including permissions, untrusted inputs, uncertain outcomes, and human approvals.
- Ability to design experiments, debug real system behavior, and explain measured improvements and remaining uncertainty.
- Ability to take ownership of ambiguous problems, collaborate effectively, and ship high-quality software.
Nice-to-haves
- Agent memory and retrieval
- Skill acquisition
- Reinforcement learning or post-training
- Trajectory datasets
- Sandboxed execution
- Distributed systems
- Inference optimization
- Multimodal models
- Voice models
Compensation
- Base salary: $250,000–$325,000 USD annually.
- Equity: 0.15%–0.30%.
- US visa sponsorship available.
Skills
Python, TypeScript, Llm Tool Use, Multi-Agent Systems, Context Engineering, Browser Automation, Computer Automation, Asynchronous Services, State Machines, Concurrency, Distributed Systems, Reinforcement Learning, Sandboxed Execution, Observability, Inference Optimization
Similar jobs
ML Engineering jobsStaff Machine Learning Engineer building and operating production ML systems for causal marketing measurement, optimization, and planning. The role requires deep statistical and machine learning expertise, production programming experience, cross-functional collaboration, and technical mentorship.
Staff engineer responsible for designing and scaling the infrastructure, execution environments, verifiers, and tooling used to train and evaluate AI agents. Requires 8+ years of software engineering experience, strong Python and distributed-systems expertise, and familiarity with sandboxing, high-throughput systems, and LLM workflows.
Senior Staff ML Engineer fine-tunes and optimizes state-of-the-art LLMs for Airbnb's customer support AI products, including AI assistants and autonomous agents. Partners cross-functionally to productionize models at scale. Requires PhD and 10+ years experience with PyTorch.
Leads the roadmap and technical vision for Snowflake Feature Store, building reliable, high-performance machine learning platform capabilities and supporting technical execution across partner teams. Requires 10+ years of experience with data-serving infrastructure or ML platforms, plus Java and Python expertise.
Leads the design, implementation, integration, and field validation of tactical autonomy and multi-agent coordination capabilities for unmanned platforms. Requires 7+ years of relevant experience, production C++, technical leadership, and eligibility for a U.S. Secret clearance.