Staff Software Engineer
Build and lead Traba's agentic platform as a founding member of the Agents team. Architect orchestration, evals, model strategy, and integrations for autonomous AI agents in industrial supply chain workflows. Requires 7+ years engineering experience including 2+ years production LLM/agent systems, strong Python/TS skills, and customer immersion.
About the job
Responsibilities
- Architect Traba's agent platform end-to-end: orchestration runtime, eval and observability stack, integration layer to internal and customer systems (WMS/TMS/ERP), and patterns every agent is built on.
- Own foundational technical decisions: model strategy, harness design, retrieval and memory architecture, tool/MCP surface, and quality measurement.
- Spend time in the field with customers and operators, translating observations into durable product and repeatable deployment patterns.
- Build evaluation as an engineering discipline: datasets, graders, regression suites, and experimentation tooling.
- Hire and mentor engineers; set the standard for quality.
- Partner with CTO, product, and ops leadership on the multi-year platform roadmap.
Requirements
- 7+ years of software engineering experience, with 2+ years of hands-on production work on LLM- or agent-based systems.
- Deep expertise in Python and/or TypeScript/Node.js.
- Track record designing distributed systems, APIs, and data models on PostgreSQL and modern messaging (Kafka, RabbitMQ, or equivalent).
- Demonstrated ownership of a non-trivial production agent system: orchestration, tool use, retrieval, evals, observability, and cost/latency tuning.
- Background in vertical AI / AI-agent company, forward-deployed engineering role, or AI-native data company.
- History of leading 0-to-1 builds in early-stage environments with high agency.
- Strong written and verbal communication skills.
Nice-to-Haves
- Supply chain, logistics, or industrial exposure.
- Experience with WMS quirks, shift cadence, exception handling.
Skills
Python, TypeScript, Node.js, Postgres, Kafka, RabbitMQ, LLMs, Agent Systems, Distributed Systems, APIs, Orchestration, Retrieval, Evals, Observability
Similar jobs
ML Engineering jobs
Snowflake
Bellevue, WA
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.
Shield AI
San Mateo, CA
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.
Airbnb
United States
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.
Harvey
San Francisco, CA
Leads the design and operation of reliable, scalable model infrastructure powering AI inference across multiple providers. Requires 7+ years of distributed-systems engineering experience, strong programming skills, and expertise in production reliability and cloud infrastructure.
Haus
Seattle, WAStaff 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.