Axiom Bio Logo

Axiom Bio

Platform Engineer

Reposted 4 Days Ago
In-Office or Remote
Hiring Remotely in San Francisco, CA, USA
Senior level
In-Office or Remote
Hiring Remotely in San Francisco, CA, USA
Senior level
Lead design and build of core infrastructure for enterprise ML: model evaluation/deployment, inference/serving, data storage/retrieval, and production deployment of large-scale reasoning agents. Work with enterprise customers, integrate on-prem systems, and mentor scientists to adopt strong engineering practices and reliability-focused ownership.
The summary above was generated by AI

About Axiom:

Axiom is building a compounding ecosystem to replace animal testing and, over time, reshape how clinical trials are run. It starts with deeply understanding the needs of drug hunters inside large pharma. Those needs shape the world-class datasets we build from scratch. We then use that data to advance our own ML research, while also collaborating with leading AI labs to improve frontier models’ ability to reason over Axiom’s data inside Axiom’s agent harness. This creates a compounding loop: deeper customer understanding shapes the data we generate; better data improves frontier models, Axiom’s fine-tuned models, and our agentic infrastructure; stronger models and tooling expand the capabilities we can offer; and those capabilities are forward deployed into pharma's drug discovery workflows, where scientists use them to solve the highest value drug discovery problems. In turn, this helps us identify the next problems to tackle. Today, we are focused on solving drug-induced liver injury through an integrated data and agentic system already being used by 7 of the top 20 pharma companies and several of the world’s most innovative biotechs. Over time, Axiom will build the world’s largest human datasets across all the major organ systems, paired with an agentic harness that uses this data to predict human drug outcomes dramatically better than animals.

What you will be doing:

  • Lead Axiom’s evolution into a world-class engineering company focused on enterprise ML software

  • Design and build the core infrastructure that powers Axiom’s enterprise ML systems, including model evaluation/deployment, model inference/serving, and customer data management

  • Architect scalable systems for inference, storage, and retrieval of chemical, biological, and clinical data

  • Deploy large-scale reasoning agents from research environments into production, integrating them into on-prem customer-facing products and infrastructure

  • Teach and empower scientists across ML, chemistry, and biology to become great engineers by instilling a great engineering culture

Various expertise which gets us interested:

  • Built SaaS products that store and process large volumes of customer data.

  • Worked directly with large enterprise customers and supported their complex software needs

  • Designed and developed large-scale machine learning systems covering data access, training, evaluation, and deployment

  • Handled the “messy” parts of ML deployment, such as evaluation pipelines, versioning, and monitoring

  • Built LLM-powered data systems, with a focus on research workflows and information retrieval

Key criteria:

  • Strong generalist software engineer with experience across cloud infrastructure,machine learning, backend systems, distributed systems

  • Enjoys working with enterprise customers and simplifying complex technical solutions to meet their needs

  • Built and deployed production systems used by large enterprise businesses

  • Invested in team growth particularly when it comes to building strong engineering culture across the company

  • Passionate about collaborating with researchers and scientists, helping them become strong engineers

  • Takes full ownership of the customer experience—deeply focused on reliability and all the ways things can go wrong

  • Demonstrates relentless

HQ

Axiom Bio San Francisco, California, USA Office

San Francisco, CA, United States, 94107

Similar Jobs

Yesterday
In-Office or Remote
97K-170K Annually
Senior level
97K-170K Annually
Senior level
Healthtech • Retail • Pharmaceutical • Telehealth
Ownership of platform reliability and incident response across cloud infrastructure. Improve observability, CI/CD pipelines, deployments, developer experience, and operational tooling. Act as primary on-call platform expert, collaborate with product teams, and run postmortems to raise platform reliability.
Top Skills: AWSBashClaude CodeCloudflareCoderabbitCodexCursorDatadogDjangoDockerEcs/FargateGCPGeminiGitGitGithub ActionsGleanHuskyPostgresPythonStripeTerraformTwilioVercel
8 Days Ago
Easy Apply
Remote
Easy Apply
181K-241K Annually
Senior level
181K-241K Annually
Senior level
Big Data • Fintech • Mobile • Payments • Financial Services
Lead design and implementation of Affirm's Lakehouse Platform (Iceberg, Spark, Snowflake). Define technical strategy, enable secure governed access, improve analytics engineering practices, optimize performance and cost, operate at scale, mentor engineers, and collaborate cross-functionally to deliver reliable, discoverable, and cost-efficient analytical data infrastructure.
Top Skills: Apache IcebergCloud-Native StorageDbtJvm (Java/Scala)PythonSnowflakeSparkSQLTerraform
8 Days Ago
Easy Apply
Remote
Easy Apply
133K-183K Annually
Junior
133K-183K Annually
Junior
Big Data • Fintech • Mobile • Payments • Financial Services
Build and operate ML training and serving infrastructure: design, develop, and launch backend systems, support availability and on-call, collaborate cross-functionally, and improve developer velocity.
Top Skills: AWSKotlinKubernetesMySQLPython

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account