Profile cover
E

Senior AI Training Job Preemption Recovery Engineer

Xivio Solutions

Remote · $200k+ / year · Full-time · Posted 1 month 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

Our customers run multi-day training jobs on spot and preemptible GPUs to save money, and they lose hours of progress every time a node disappears. Xivio Solutions already handles capacity planning and scheduling; the missing piece is fast, automatic recovery. You'll build it: asynchronous checkpointing that doesn't stall training, elastic resumption on a different number of GPUs, and smart rescheduling across AWS and Azure capacity pools. You'll work with a team of five infra engineers and be the technical owner of recovery end to end.

Fully remote.

Responsibilities

  • Design asynchronous, sharded checkpointing for large distributed training jobs
  • Build elastic resume so jobs restart on whatever capacity is available
  • Own recovery time and wasted-GPU-hours metrics across all customer clusters
  • Drive adoption with customer ML teams through docs, examples and office hours

Requirements

  • Significant experience with distributed training (FSDP, DeepSpeed or similar)
  • Strong Kubernetes and cloud infrastructure background on AWS or Azure
  • Comfortable debugging NCCL, storage throughput and scheduler issues
  • History of owning reliability-critical systems

Benefits

  • Remote in the US
  • Salary at the top of market plus equity
  • Unlimited PTO with a 4-week minimum
  • Premium health coverage
  • Annual hardware refresh

Keep exploring related roles

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

Categories

MLOps & AI Infrastructure

Skills

python kubernetes pytorch aws azure

Related remote AI jobs

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

Browse all jobs

Senior Software Engineer - Behavioral Security Products

Abnormal Security

Full-time Senior

Lead Technical Onboarding Engineer

Neo4j

Full-time Lead

Staff Software Engineer, Storage Platform

Wayve

Full-time Staff

Senior Data Platform Engineer - Weights & Biases

CoreWeave

Full-time Senior

Staff Software Engineer, AI Libraries

Wayve

Full-time Staff

Adoption Specialist - Infrastructure & Agentic

Cohere

Full-time

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$200k+ / year

AI Expertise

MLOps & AI Infrastructure

Skills

python kubernetes pytorch aws azure docker

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 profile and 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