Profile cover
C

AI Experiment Compute Cost Attribution Developer

Gravex Cloud

Hybrid · $130k – $160k / year · Contract-to-hire · Posted 3 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 serve ML teams at companies running hundreds of experiments in parallel, and right now the only cost visibility they get is a single lump AWS bill at the end of the month. Our model registry and feature store already track model and feature lineage well, but nobody can tell a team lead which experiment, which researcher, or which dataset actually drove last month's GPU spend. We're looking for a senior engineer to build cost attribution that ties every dollar of compute back to a specific experiment run, so teams can finally see where their budget goes. This starts as a contract engagement with a clear path to a full-time offer for the right person.

Responsibilities

  • Build cost attribution linking AWS compute spend to individual experiment runs
  • Integrate attribution data into the existing model registry and feature store
  • Work with MLflow to tag experiment metadata needed for cost tracing
  • Investigate historical spend to validate attribution accuracy
  • Propose budget alerts based on per-team or per-project spend trends

Requirements

  • 5+ years of experience with ML infrastructure or cloud cost engineering
  • Experience with MLflow or similar experiment tracking tools
  • Strong SQL and data pipeline skills
  • Familiarity with AWS billing and cost allocation tooling
  • Comfortable working independently on a contract basis initially

Benefits

  • Hybrid role, flexible office attendance in Austin
  • Competitive contract rate with path to full-time
  • Health insurance available after conversion
  • Direct collaboration with senior ML infrastructure engineers
  • Opportunity to shape long-term cost strategy

Keep exploring related roles

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

Categories

MLOps & AI Infrastructure

Skills

python mlflow docker aws sql

Related remote AI jobs

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

Browse all jobs

AI Security Engineer Intern (Summer 2027)

RemoteAI company

Full-time Intern

Software Engineer, Infrastructure

RemoteAI company

Full-time

Senior Infrastructure Engineer

RemoteAI company

Full-time Senior

Senior Infrastructure Engineer

RemoteAI company

Full-time Senior

Revenue Operations Manager

RemoteAI company

Full-time

Software Engineer - AI Tooling Infrastructure

RemoteAI company

Full-time

Job Type

Contract-to-hire

Level

Senior

Language

English

Salary Range

$130k – $160k / year

AI Expertise

MLOps & AI Infrastructure

Skills

python mlflow docker aws sql

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