Profile cover
J

Senior ML Engineer

Axion Labs

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

No longer accepting applications

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

We're building the next generation of AI-powered drug discovery tools and need a senior ML engineer who's comfortable working at the intersection of deep learning and biochemistry. Our platform processes millions of molecular simulations daily, and the models behind them need constant iteration. You'll own the full lifecycle of our prediction models — from prototyping new architectures in PyTorch to deploying them behind FastAPI endpoints on Kubernetes. The team is small (8 engineers) but ships fast: we release weekly and run continuous A/B tests on model outputs against wet-lab results. This isn't a research-only role. Every model you build will be tested against real experimental data within weeks. If you want your work to have tangible impact on how new therapeutics are discovered, this is the place.

Fully remote.

Responsibilities

  • Own the full ML pipeline from data ingestion to production deployment
  • Architect and iterate on deep learning models for molecular property prediction
  • Partner with the biotech team to validate model outputs against lab data
  • Build and maintain model serving infrastructure on Kubernetes
  • Drive adoption of experiment tracking and reproducibility best practices

Requirements

  • 5+ years building production ML systems
  • Deep expertise in PyTorch and model optimization
  • Experience deploying models at scale with Docker and Kubernetes
  • Familiarity with MLflow or similar experiment tracking tools
  • Bonus: background in computational biology or chemistry

Keep exploring related roles

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

Categories

AI & Machine Learning Engineers MLOps & AI Infrastructure

Skills

python pytorch docker kubernetes mlflow

Related remote AI jobs

Nearby roles that share similar skills, categories, or hiring intent.

Browse all jobs

Research Engineer / Scientist, Pre-training

Anthropic

Full-time

AI Security Engineer Intern (Summer 2027)

Brainco

Full-time Intern

Software Engineer, Infrastructure

Niantic Spatial

Full-time

Research Engineer, Infrastructure

Harvey

Full-time

Research Engineer, Infrastructure

Harvey

Full-time

Software Engineer - AI Tooling Infrastructure

Supabase

Full-time

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$160k – $200k / year

AI Expertise

AI & Machine Learning Engineers MLOps & AI Infrastructure

Skills

python pytorch docker kubernetes mlflow fastapi

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
Create free account & apply Log in

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