Profile cover
J

AI Biomarker Discovery Platform Engineer

Axion Labs

Remote · $160k – $200k / year · Full-time · Posted 5 months ago

This role is no longer accepting applications

Don't miss the next role like this

Create a free profile with your skills. We match you to new AI roles as they open, and companies hiring can contact you directly.

Create free profile, 2 minutes

Similar open roles you can apply to now:

See all open AI jobs

About the role

Drug discovery pipelines generate petabytes of multi-omics data — genomics, proteomics, metabolomics — but the biomarkers that predict therapeutic response hide in the noise. Our platform uses deep learning to find them, reducing the biomarker identification phase from 18 months to 6 weeks for our pharmaceutical partners. You'll own the model development pipeline: designing architectures that handle heterogeneous biological data types, building feature selection methods that produce clinically interpretable biomarker panels, and validating discoveries against independent patient cohorts. The models must satisfy both statistical rigor (p-values that survive Bonferroni correction) and clinical utility (biomarkers that lab technicians can actually measure). The computational biology team is seven engineers and three PhD scientists. You'll be the senior ML engineer responsible for architecture decisions, experiment design, and the production pipeline that delivers validated biomarker candidates to our pharma clients' wet labs.

Fully remote.

Responsibilities

  • Design deep learning architectures for multi-omics biomarker discovery
  • Build feature selection pipelines producing clinically interpretable biomarker panels
  • Validate biomarker candidates against independent patient cohorts
  • Architect the production pipeline from raw data to validated candidates
  • Collaborate with bioinformatics scientists on domain-specific model constraints
  • Present findings to pharma partner scientific advisory boards

Requirements

  • 6+ years in ML engineering with biomedical or genomics data experience
  • Experience with high-dimensional feature selection under multiple testing correction
  • Strong PyTorch and distributed computing skills
  • Understanding of biological data types and their statistical properties
  • Track record of shipping ML systems in regulated environments

Benefits

  • Fully remote with quarterly team meetings
  • Top-of-market compensation plus equity
  • Comprehensive health coverage
  • $5,000 annual conference and learning budget
  • Direct impact on drug discovery timelines

Keep exploring related roles

Use these topic pages to discover more roles in the same hiring cluster.

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$160k – $200k / year

AI Expertise

AI & Machine Learning Engineers Data Science & Analytics

Skills

python pytorch spark sql docker aws

Ready to apply for this role?

Create a free talent account.

  • Find AI roles in one place
  • Get AI-matched job recommendations
  • Reply to companies that message you
  • Build your public AI talent profile

This role is closed. Create your free profile to apply to similar open roles.

Don't miss the next one: get matched to new roles like this.

Create free profile