Platform Engineer

ABOUT AXIOM Axiom is building the translational intelligence layer for drug discovery: AI systems that help scientists predict human toxicity earlier, more accurately, and more mechanistically than animal studies or legacy in vitro assays. Unexpected toxicity is one of the largest reasons drug programs fail. Today, drug discovery teams still rely on fragmented assays, animal studies, and expert judgment to decide which molecules are safe enough to advance. We believe this can be dramatically improved. At Axiom, we generate and curate massive multimodal datasets spanning chemical structures, primary human cell imaging, multicellular tissue systems, transcriptomics, proteomics, mass spectrometry, ADME, dose-response curves, clinical outcomes, and human exposure. To date, we have built the largest experimental-to-clinical dataset in the world and we are just getting started. We use these datasets to train models and agents that connect chemistry, biology, mechanism, and clinical risk. We are looking for an infrastructure / platform engineer to build the systems that make this work at scale. You will own the backend, distributed systems, model-serving infrastructure, deployment pipelines, customer data systems, and enterprise platform architecture behind Axiom’s AI products. This is a role for a deeply technical generalist who wants to help Axiom evolve into a world-class engineering company. CHARTER Build the infrastructure that powers the first scientific AI systems capable of replacing animal and legacy toxicity experiments. You will create the platform that turns Axiom’s research into reliable, secure, enterprise-ready software used by the world’s leading drug discovery teams. WHAT YOU WILL DO You will own critical systems across Axiom’s backend, ML platform, customer deployment, and enterprise infrastructure. You will: - Lead Axiom’s evolution into a world-class engineering organization focused on enterprise ML and data software. - Design and build the core infrastructure powering Axiom’s ML systems, including model evaluation, model deployment, inference, serving, monitoring, and versioning. - Architect scalable systems for storing, retrieving, processing, and serving chemical, biological, clinical, customer, and model-generated data. - Deploy large-scale reasoning agents from research environments into production systems used by customers. - Build infrastructure for running image models, LLM agents, mechanistic reasoning systems, and multimodal toxicity models at scale. - Create robust systems for customer data management, including secure ingestion, access control, audit trails, versioned deliveries, and customer-specific workspaces. - Build the backend systems behind Axiom’s product, including APIs, data services, inference services, workflow systems, and internal tooling. - Support enterprise customer deployments, including cloud, secure VPC, and potentially on-prem or customer-controlled environments. - Build evaluation and observability systems for ML models and agents, including regression testing, model comparison, trace inspection, rollout monitoring, and failure analysis. - Work with ML researchers to turn prototypes into reliable production systems. - Work with scientists to turn research workflows into durable software. - Work with product and customer teams to ensure enterprise users can trust, understand, and depend on Axiom’s systems. - Teach and empower scientists, ML researchers, and engineers to write better software and build better systems. - Help define Axiom’s engineering culture from the ground up. WHAT WE ARE LOOKING FOR We are looking for a strong generalist software engineer with excellent taste in systems, infrastructure, and product. You might be a great fit if: - You have built production systems used by large enterprise customers. - You have designed backend or distributed systems that process large amounts of data reliably. - You have built SaaS products that store, process, and serve sensitive