Skip to content

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, plus customer immersion and 0-to-1 leadership.

240k – 300kNew York, NYSan Francisco, CAML EngineeringHybrid7+ YOE

About the role

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, 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.
  • Strong written and verbal communication skills.

Nice-to-Haves

  • Supply chain, logistics, or industrial exposure.
  • Experience with WMS quirks, shift cadence, exception handling.

Skills

PythonTypeScriptNode.jsPostgresKafkaRabbitMQLLMsAgent SystemsDistributed SystemsAPIsOrchestrationRetrievalEvalsObservability

Similar roles

ML Engineering jobs

Staff Software Engineer

Build and scale Traba's applied AI platform by integrating frontier models and agents into production systems for automating industrial staffing pipelines. Requires 7+ years deep full-stack experience with TypeScript/Node.js or Python, PostgreSQL, messaging systems, and distributed systems; AI/LLM production experience a plus.

240k – 300kNew York, NY +1ML EngineeringOn-site7+ YOEAIAPIs

Staff AI/ML Engineer

Build and productionize AI/ML systems including recommendations, NLP interfaces, and agentic workflows. Requires 10+ years building production-grade AI/ML systems and experience with foundation models.

240k – 270kNew York, NYML EngineeringOn-site10+ YOEAWSGCP

Staff AI/ML Engineer

Build and productionize AI/ML systems including recommendations, NLP interfaces, and agentic workflows. Requires 10+ years building production-grade AI/ML systems and experience with foundation models.

240k – 270kSan Francisco, CAML EngineeringOn-site10+ YOEAWSGCP

Staff Applied AI Engineer, AI Experiences & Agents

Build and ship production-grade AI agents and experiences that help creators personalize, automate, and grow. Requires strong product engineering skills, hands-on LLM/agent experience, and rigorous evaluation practices.

240k – 250kLos Angeles, CA +1ML EngineeringOn-siteRAGLLMs

Staff AI Engineer

Staff AI Engineer shapes technical direction of AI agents for audit workflows, owning architecture, LLM infrastructure, and reliability systems. Requires 8+ years experience with deep expertise in TypeScript/Python, RAG, and enterprise-scale LLM products.

240k – 310kSan Francisco, CAML EngineeringHybrid8+ YOERAGLLMs