Profile cover
C

AI Experiment Tracking UI Backend Developer

Gravex Cloud

Remote · $60k – $80k / 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

Data scientists run hundreds of experiments per week — different hyperparameters, architectures, training data subsets, preprocessing steps — and keeping track of what was tried, what worked, and what didn't is a nightmare. Our experiment tracking platform stores every run's configuration, metrics, artifacts, and lineage in a searchable, comparable interface. As a junior developer, you'll build the backend APIs that power the experiment tracking UI: endpoints for querying runs by metric ranges, comparing multiple experiments side-by-side, grouping runs into logical experiments, and generating summary reports for team reviews. The data is already stored in MLflow; your job is building the API layer that makes it useful for the ways data scientists actually work — filtering, sorting, comparing, and annotating experiments. The developer tools team is four engineers. You'll own the API layer, working with the frontend developer who builds the comparison views and the platform engineer who manages the MLflow infrastructure.

Fully remote.

Responsibilities

  • Build API endpoints for experiment querying, filtering, and comparison
  • Design experiment grouping and annotation features
  • Implement side-by-side metric comparison across multiple runs
  • Create summary report generation for team experiment reviews
  • Optimize query performance for users with thousands of tracked runs
  • Write API documentation and integration guides

Requirements

  • 1+ year of Python backend development with API design
  • Familiarity with SQL for data querying and aggregation
  • Basic understanding of ML experiment workflows
  • Experience with FastAPI or similar web frameworks
  • Interest in developer tools and ML infrastructure

Benefits

  • Fully remote across EU time zones
  • 28 days PTO
  • €1,500 learning budget
  • Structured mentorship from senior engineers
  • Build tools the ML community relies on daily

Keep exploring related roles

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

Categories

MLOps & AI Infrastructure Automation & AI Workflows

Skills

python sql fastapi docker mlflow

Related remote AI jobs

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

Browse all jobs

AI Security Engineer Intern (Summer 2027)

Brainco

Full-time Intern

Software Engineering Manager, AI Migrations

Snowflake

Full-time

Software Engineer, Infrastructure

Niantic Spatial

Full-time

Senior Infrastructure Engineer

Harvey

Full-time Senior

Senior Infrastructure Engineer

Harvey

Full-time Senior

Revenue Operations Manager

Handshake

Full-time

Job Type

Full-time

Level

Junior

Language

English

Salary Range

$60k – $80k / year

AI Expertise

MLOps & AI Infrastructure Automation & AI Workflows

Skills

python sql fastapi docker mlflow 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
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