Senior/Staff Software Engineer - ML/AI Applications

Zoox
$225K - $249K/yr

AI Infrastructure

Tech Stack

About the Role

As a Technical Lead on Zoox's internal agentic platform, you will own end-to-end integration of a tool-centric agent runtime with external and internal agent clients, line of business backends, scheduled and interactive agent workloads, and cloud operations (Kubernetes, observability, secrets management). You will translate architecture and security intent into stable tool contracts, testable integration milestones, and repeatable releases, and you will be the primary engineer when issues sit at the boundary between product surfaces, platform services, and upstream systems.
This role is platform and service integration for an agentic framework that mediates tools (discovery, invocation, policy, and telemetry). Success is measured by reliable tool call paths over HTTP, correct authentication and scoping, environment parity from dev through production, and clear ownership across partner engineering, security, and infrastructure teams.

In this role, you will::

Own and lead integration strategy and execution across hosted services, shared libraries, standardized tool protocols (e.g., JSON-RPC over HTTP), REST APIs, and diverse callers (vendor agents, consoles, automation workers).
Drive cross-functional delivery with backend API teams, partner agent owners, security, SRE/platform, and program management, ensuring formal sign-off where required.
Debug and root-cause complex failures using structured logs, distributed traces, metrics, and gateway/session-token behavior, documenting durable mitigations and runbooks.
Define and review all integration artifacts (contracts, allowlists, registry config, environment matrices, and interface assumptions) that upstream and downstream teams implement against.
Contribute to release gating by ensuring container image promotion, configuration/secrets posture, smoke path execution for high-traffic flows, and disciplined branching/tagging for production intent

Qualifications::

BS or MS in Computer Science or a related field with 7+ years of software engineering experience, strong in integration, platform, or backend roles.
Strong Python (production services, async I/O, automated testing) and comfort in a large monorepo with CI-driven delivery.
Deep experience with HTTP integration (REST/JSON-RPC, TLS, debugging under load) and shipping services on Kubernetes (deployments, health checks, troubleshooting).
Auth integration experience: bearer tokens, JWT validation, OAuth2/OIDC or enterprise SSO concepts, and least-privilege scoping

Bonus Qualifications:

Exposure to Model Context Protocol (MCP) or similar agent tool-calling and orchestration models used in industry or internal platforms.
Observability-led integration habits: metrics, traces, structured logging, and basic SLO or dashboard thinking for integration health.
Background in security-sensitive integrations (data minimization, external partner access, review cycles with security or compliance).
About Zoox
Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.
Follow us on LinkedIn
Accommodations
If you need an accommodation to participate in the application or interview process please reach out to accommodations@zoox.com or your assigned recruiter.

A Final Note:

You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills.

Apply Now
Apply Now

More jobs like this

Docker logo
Staff Software Engineer, Agentic Platform

Docker

$170K - $276K/yrMCP
🇺🇸
2026-07-20
Writer logo
Software engineer, generative AI

Writer

$112K - $304K/yrPineconeWeaviate+1 more
🇺🇸
2026-07-20
Saviynt logo
Principal Software Engineer, AI Onboarding

Saviynt

$240K - $250K/yrGoogle ADKLangChain+2 more
🇺🇸
2026-06-28
Bunkerhill Health logo
AI Product Engineer

Bunkerhill Health

$160K - $260K/yr
🇺🇸
2026-06-28
Trase Systems logo
Staff Software Engineer (Product + Platform Development)

Trase Systems

$180K - $240K/yr
🇺🇸
2026-07-27
HackerOne logo
Staff Software Engineer, Applied AI

HackerOne

$230K - $280K/yrAutoGenSemantic Kernel+2 more
🇺🇸
2026-07-20
Level AI logo
Senior Machine Learning Engineer

Level AI

FAISSHugging Face+3 more
🇮🇳
2026-06-28
Databricks logo
Staff Fullstack Engineer, Agentic Applications

Databricks

$192K - $260K/yrAutoGenCrewAI+5 more
🇺🇸
2026-07-20
Klaviyo logo
Senior Software Engineer, Agent Platform

Klaviyo

$148K - $222K/yrLangChainArize+6 more
🇺🇸
2026-07-20
Metropolitan Commercial Bank logo
AI Engineer

Metropolitan Commercial Bank

$130K - $200K/yrLangChainLlamaIndex+2 more
🇺🇸
2026-06-28
NextGen Federal Systems logo
MCP/Integration Engineer

NextGen Federal Systems

MCP
🇺🇸
2026-06-13
NextGen Federal Systems logo
Application/Agentic AI Engineer

NextGen Federal Systems

AutoGenCrewAI+3 more
🇺🇸
2026-06-13
Sema4.ai logo
Staff Engineer, Agentic Backend

Sema4.ai

🇺🇸
2026-07-20
Vendelux logo
Staff AI Engineer

Vendelux

LangChainHugging Face
🇺🇸
2026-06-24
Zuma logo
Staff Engineer (Agentic AI & Data)

Zuma

🇺🇸
2026-06-24
Inflection AI logo
Staff Engineer, Agentic

Inflection AI

$350K - $500K/yrRemoteClaude Agent SDKLangGraph+4 more
🇺🇸
2026-06-24
Paytm logo
Senior AI Engineer

Paytm

AWS BedrockAzure OpenAI+1 more
🇨🇦
2026-07-07
BlueCat logo
AI Infrastructure Engineer – BlueCat Horizon Platform

BlueCat

CA$160K - CA$180K/yrLangChain
🇨🇦
2026-06-28
Granted logo
Backend Engineer, Agentic AI (Staff+ Level)

Granted

$150K - $225K/yrRemote
🇺🇸
2026-06-24
Everlaw logo
Senior Software Engineer, AI Platform

Everlaw

$173K - $251K/yr
🇺🇸
2026-07-20

Explore related roles

Get jobs like this weekly

Join 84 subscribers