Skip to content
AxionAxionSan Francisco, CA

Staff Software Engineer, Agentic Platform

Senior individual contributor architecting and scaling agentic LLM systems that turn messy manufacturing data into reliable root-cause insights. Owns orchestration, retrieval, evaluation, and guardrails for non-deterministic production systems.

250k – 270k
Hybrid7+ YOEML Engineering

About the role

Key Responsibilities

  • Own technical architecture for agentic systems—orchestration, tool use, memory, retrieval, evaluation, observability, and guardrails
  • Build and harden the reasoning layer and infrastructure that take customer data from ingestion through agent-driven reasoning to root-cause insight
  • Establish rigorous evaluation and testing for non-deterministic systems—measuring agent accuracy against expert-validated ground truth
  • Partner across Engineering and Product to shape the technical vision and roadmap for the agent platform
  • Raise engineering standards through code review, design docs, and technical direction
  • Mentor engineers on agentic patterns and system design

Requirements

  • Minimum 7+ years of software engineering experience, with recent, hands-on focus on building agentic systems (LLM orchestration, tool/function calling, multi-step and multi-agent reasoning, retrieval and context engineering, and interoperability protocols such as MCP)
  • Demonstrated ability to design and own complex distributed systems, ideally in data-intensive B2B SaaS environments
  • Hands-on fluency with the modern agentic stack and strong intuition for failure modes of non-deterministic, LLM-driven systems
  • Experience turning messy, heterogeneous real-world data into something a model can reason over reliably
  • Track record of technical mentorship
  • Strong executor in fast-paced environments who can balance shipping MVPs with production-grade systems

Nice-to-Haves / Culture Fit

  • Thrives in early-stage companies
  • "Data First" philosophy
  • Comfortable with ambiguity and fast-moving situations

Skills

Llm OrchestrationTool/Function CallingMulti-Agent ReasoningRetrieval SystemsContext EngineeringMcpDistributed SystemsObservabilityEvaluation FrameworksData-Intensive Systems

Similar roles

ML Engineering jobs
The Voleon Group

Member of Research Staff, Optimization

The Voleon GroupBerkeley, CA +1

Conduct optimization research and implement large-scale constrained optimization models that drive real-time trading decisions, working across the full research lifecycle from theory to production. Requires PhD-level coursework and strong applied research background in optimization.

250k – 275k
Hybrid7+ YOEML Engineering
Nuance Labs

Member of Technical Staff — RL Research

Nuance LabsSeattle, WA

New/recent PhD to own RL and post-training for large-scale omni models. Build and scale the full RL/post-training stack including rollout, optimization, reward modeling, and evaluation for real-time audiovisual AI.

250k – 350k
On-siteEntry levelML Engineering
Labelbox

Staff Software Engineer, AI Data Platform

LabelboxSan Francisco, CA

Staff-level engineer building AI data platform infrastructure, eval systems, and agent-first tooling for frontier labs. Requires 4+ years shipping production systems, full-stack experience, and deep TypeScript/Python proficiency.

250k – 280k
Hybrid4+ YOEML Engineering
Nuance Labs

Member of Technical Staff — Model Optimization and Inference

Nuance LabsSeattle, WA

Optimize inference for real-time multimodal AI avatars. Specialize in LLM and diffusion model serving, KV cache strategies, quantization, and low-latency frameworks like vLLM and TensorRT-LLM.

250k – 350k
On-site7+ YOEML Engineering
ClickUp

Staff AI Engineer - AI Product

ClickUpUnited States

Leads development of user-facing AI features using LLMs and AI models, integrating them into production for scalable, personalized experiences. Requires 5+ years engineering experience with Python/JS, databases, and AI orchestration expertise.

250k – 300k
Remote5+ YOEML Engineering