Agent Harness Engineer

About Axiom:

Axiom is building the closed-loop scientific AI system required to replace animal testing and, over time, much of human safety testing. We start with pharma’s hardest drug development toxicology problems. Those problems define the proprietary human biological data we generate through Axiom’s Data Factory. We use that data to train scientific AI, partnering with leading AI labs to improve frontier models while building our own specialist agentic harness to deploy the improved frontier models back into pharma. Each deployment reveals the next capabilities to build, creating a compounding loop across data, models, and drug development. Today, liver toxicity is our proving ground. Axiom is already helping leading pharmaceutical companies understand toxicity, identify its mechanism, and design safer drugs. Over time, we will expand across the major organ systems and build the experimental and agentic system of record for translational drug development. Our goal is to dramatically reduce the risk of testing new molecules in humans, enabling high throughput evaluation of efficacy in humans.

What you will do:

Own the harness: the scaffolding, tooling, and infrastructure that turn frontier models into agents that do long-horizon scientific analysis

Build the data backbone that gets everything to the right place: pipelines, storage, and systems for runtime context, agent trajectories, eval results, and training data

Build sandboxed execution environments with instant spin-up/tear-down, reproducible and deterministic enough to trust for evals and RL

Design and run the eval systems: offline suites, test cases on production traces, LLM-as-judge pipelines, regression gates

Work closely with domain experts and encode their taste into rubrics, golden sets, and review workflows, turning "I know it when I see it" into something measurable

Build the tools the agent needs to do better work, and stay tuned in to the state of the art for new methods, protocols, and patterns worth adopting

Make every agent run observable and replayable: trace every model call, tool call, and state transition, and build the debugging tooling to make sense of it

Engineer the context: memory, compaction, retrieval, and recovery so long-horizon agent runs stay coherent across hours and crashes

Own the loop itself: retries, budget caps, stop conditions, output verification, permissions, and guardrails

Support ML research with environments, reward instrumentation, and rollout infra for RL on agentic tasks

Expertise:

Python, Modal, DuckDB, FastAPI, Docker, Containerization, Terraform

Engineers who've built with LLM APIs and shipped agentic systems: tool use, loops, and the debugging scars to prove it

Built bespoke evaluation, monitoring, and RL env observability tooling (SvelteKit, Svelte 5, React)

What we look for:

Can tackle deep technical challenges and own/ship simple, clean, maintainable code

High ownership: owns outcomes end to end, not tickets, and doesn't wait for a spec to start moving

Allergic to complexity: reaches for the simplest system that works and keeps it that way as it scales

Strong software engineer first, with infrastructure, platform, data, or devtools depth and production systems they're proud of

Instinctively asks "how would we know if this is working?" and builds the measurement alongside the feature

Reads an agent failure trace the way other engineers read a stack trace

Cares about reliability because they know environments that break silently poison evals and training data

Has a knack for surfacing the important questions about what the agent actually needs to do better work

Sharp and confident

keeps up with a really technical & sophisticated crowd

comfortable getting in over their head and figuring it out as they go

thrives in a discipline with no playbook, because it's being invented right now

Curious about how things work: an engineering/tinkering mindset, good at scavenging the state of the art

Passion for learning what "good" looks like from deep domain experts and turning it into systems