Axiom Bio Logo

Axiom Bio

Platform Engineer

Reposted Yesterday
In-Office or Remote
Hiring Remotely in India
Senior level
In-Office or Remote
Hiring Remotely in India
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

Similar Jobs

5 Days Ago
Easy Apply
Remote
Easy Apply
Senior level
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
5 Days Ago
Easy Apply
Remote
Easy Apply
Junior
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
8 Days Ago
Easy Apply
Remote
Easy Apply
Senior level
Senior level
Big Data • Fintech • Mobile • Payments • Financial Services
Build, operate, and extend Affirm's data platform and application tooling to enable self-service data applications at scale. Implement platform features (provisioning, CI/CD, access control), integrations, data governance, automation, observability, and on-call operations. Contribute to roadmap for semantic layers, agentic data tooling, and AI-native workflows while raising engineering and operational standards.
Top Skills: Agentic Data ToolsBigQueryBuildkiteCi/CdDatabricksDockerGithub ActionsLlmsOauthOidcPythonRbacRest ApisSecrets ManagementSemantic LayerSnowflakeSQLWebhooks

What you need to know about the Pune Tech Scene

Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account