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