Axiom Bio Logo

Axiom Bio

Platform Engineer

Reposted 2 Months Ago
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 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 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

2 Days Ago
Easy Apply
Remote
Easy Apply
Junior
Junior
Big Data • Fintech • Mobile • Payments • Financial Services
Build and ship full-stack web applications and scalable APIs for Affirm’s advertising platform. Break projects into deliverable phases, collaborate with product, design, and analytics partners, write tested and extensible code, monitor service metrics, support on-call operations, review code, and contribute to engineering growth and hiring activities.
Top Skills: APIsDeclarative Ui FrameworksKotlinPythonReactWeb Applications
2 Days Ago
Remote
Senior level
Senior level
Artificial Intelligence • Consumer Web • Digital Media • Information Technology • Social Impact • Software
Build evaluation infrastructure, observability tooling, diagnostic systems, datasets, annotation workflows, and experiments for production AI agents. Lead a hands-on AI Quality engineering team, improve agent quality, latency, and cost, and partner with AI Core engineering. The role requires developing evaluation pipelines, scorers, dashboards, CI/CD systems, and model-testing frameworks while managing team priorities and technical direction.
Top Skills: BraintrustCi/CdLangsmithPythonRuby On Rails
3 Days Ago
Remote or Hybrid
Entry level
Entry level
Cloud • Fintech • Information Technology • Machine Learning • Software
Lead development of highly available, distributed payments platform services and global bill payment capabilities. Design scalable, secure, high-throughput backend systems, APIs, and database solutions while integrating multiple payment providers and risk controls. Contribute hands-on production code and proofs of concept, establish technical direction, mentor engineers, collaborate cross-functionally, and promote AI tools that improve engineering productivity.
Top Skills: Api DesignArtificial IntelligenceAWSC#C++Distributed SystemsGoJavaNoSQLSQL

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