Skip to content
WatershedWatershedSan Francisco, CA

Staff Software Engineer, AI Platform

Technical leader building agent infrastructure, observability, evals, and guardrails for production AI systems at Watershed. Requires 6+ years backend/platform/AI engineering experience and production TypeScript systems.

202k – 255k/yr
On-site6+ YOEML Engineering

About the role

Responsibilities

  • Design and build the agent infrastructure that powers Watershed's products
  • Develop the observability and tracing layer for agent decisions, making it possible to debug, evaluate, and improve agent behavior at scale
  • Build evals, harnesses, and guardrails that turn agent capabilities into production-grade, dependable systems
  • Collaborate with product and other AI engineering teams to set product and technical strategy, and define the boundaries between autonomous agent behavior, deterministic code, and human oversight
  • Keep up with developments and state-of-the-art in AI and agent infrastructure to determine what is relevant to Watershed
  • Work closely with Watershed product teams to contribute your expertise to build agent experiences across the product
  • Write performant, well-crafted, tested, and maintainable code across our technical stack

Requirements

  • 6+ years of experience in backend, platform, or AI/ML engineering
  • Experience building products and infrastructure that leverage LLMs, embeddings, and other ML technologies
  • Full lifecycle experience building, deploying, and monitoring production systems that depend on LLMs or other ML technologies
  • Experience with model evaluation, agent observability, and making non-deterministic systems reliable
  • Experience building and operating production Typescript systems
  • Must be willing to work from an office 4 days per week

Skills

TypeScriptLLMsEmbeddingsMachine LearningAgent InfrastructureObservabilityModel EvaluationBackend EngineeringProduction SystemsAi/Ml Engineering

Similar roles

ML Engineering jobs
Drata

Staff AI Engineer

DrataSan Francisco, CA

Staff AI Engineer designs and owns end-to-end production AI systems including LLM pipelines, RAG, and agentic workflows for compliance platforms. Requires 10+ years software engineering with 3+ years in ML/AI, strong Python, and production AI experience.

201k – 272k/yr
Remote10+ YOEML Engineering
Clear Street

Senior / Staff AI Platform Engineer

Clear StreetUnited States

Build and own the core AI platform powering an AI-native trading copilot. Develop high-performance Rust backend for streaming, tool execution, and safe trading actions; design robust APIs with observability and security. Requires 8+ years systems programming experience.

200k – 350k/yr
Remote8+ YOEML Engineering
Clear Street

Senior / Staff AI Model Engineer

Clear StreetUnited States

Own reliability and quality for an AI copilot in a trading platform. Design evaluation systems, benchmarks, quality gates, model improvement loops, and AI monitoring for correctness, safety, and performance in market analysis and trading workflows. Requires 8+ years production software experience and strong ML eval expertise.

200k – 350k/yr
Remote8+ YOEML Engineering
Decagon

Staff Software Engineer, Agents

DecagonSan Francisco, CA

Build and own end-to-end AI agents for enterprise customers, integrating latest text/voice models and iterating based on real-world usage. Requires 8+ years of software engineering experience with Python and TypeScript.

200k – 400k/yr
On-site8+ YOEML Engineering
Nuance Labs

Member of Technical Staff - Research Fellow

Nuance LabsSeattle, WA

3-month research fellowship for early-career researchers working on frontier Multimodal LLMs, generative modeling, and real-time audiovisual AI. Own a research problem in pretraining, post-training, RL, evaluation, or multimodal modeling. Strong PyTorch and first-author tier-1 paper required.

200k – 250k/yr
On-siteML Engineering