Profile cover
O

AI Batch Inference Scheduling Engineer

Emrex AI

Remote · $160k – $200k / year · Full-time · 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

Kubernetes, Docker, and MLflow are the core of what we build at Emrex AI. If you've optimized batch inference scheduling under real cost and latency constraints, this role should be familiar territory. You'll build the scheduler that decides how batch inference jobs get queued and prioritized across our GPU fleet, balancing client SLAs against cost efficiency — right now that logic is a set of static priority rules that don't adapt well when load spikes. You'll join a cross-functional squad with defined sprints, clear ownership, and weekly syncs with leadership, working alongside the team that already owns our fine-tuning and inference cost optimization infrastructure.

Responsibilities

  • Build a batch inference scheduler balancing SLA guarantees and cost efficiency
  • Replace static priority rules with adaptive, load-aware scheduling logic
  • Integrate scheduling with existing Kubernetes and MLflow infrastructure
  • Monitor and tune scheduler performance under real production load
  • Partner with the inference cost optimization team on shared infrastructure

Requirements

  • Strong Kubernetes and Docker experience with production ML workloads
  • MLflow or equivalent experiment and deployment tracking experience
  • AWS infrastructure experience at scale
  • Experience with scheduling or resource allocation systems
  • Comfort working in a structured, sprint-based team

Benefits

  • Fully remote
  • Equity
  • Health and dental insurance
  • 401k matching
  • Conference and training budget

Keep exploring related roles

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

Categories

MLOps & AI Infrastructure

Skills

python docker kubernetes aws mlflow

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

Principal Engineer, Data Platform

RemoteAI company

Full-time Principal

Software Engineer - AI Tooling Infrastructure

RemoteAI company

Full-time

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$160k – $200k / year

AI Expertise

MLOps & AI Infrastructure

Skills

python docker kubernetes aws mlflow

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