AI Training Infrastructure Engineer

Openai

United States

Remote

USD 150,000 - 190,000

Full time

14 days+

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Job summary

United States Digital Space LLC is seeking an AI Systems Engineer to scale infrastructure behind training and evaluation workflows. You will own projects from bottleneck identification to deployment, collaborating with researchers to improve reliability, throughput, and resource efficiency.

The role combines distributed systems engineering, performance optimization, and building self-service tools to empower researchers to launch experiments with minimal manual intervention.

Qualifications

  • Experience building or operating large-scale distributed systems.
  • Experience in ML infrastructure, inference systems, GPU performance, or infrastructure tooling.

Responsibilities

  • Build and operate infrastructure for large-scale training and evaluation, improving reliability, throughput, and resource efficiency.
  • Develop shared inference and grading platforms with automated capacity management, health monitoring, and visibility into performance.
  • Improve compute scheduling and resource allocation to reduce idle GPU time and help workloads recover quickly from failures.
  • Diagnose bottlenecks across training, inference, and orchestration, and work across teams to improve end-to-end performance.
  • Build self-service tools, automated validation, and observability that help researchers launch experiments, diagnose issues, and compare results with less manual intervention.

Skills

Distributed systems
Software engineering
ML infrastructure
GPU performance

Job description

United States Digital Space LLC is seeking an AI Systems Engineer to scale infrastructure behind training and evaluation workflows. You will own projects from bottleneck identification to deployment, collaborating with researchers to improve reliability, throughput, and resource efficiency.

The role combines distributed systems engineering, performance optimization, and building self-service tools to empower researchers to launch experiments with minimal manual intervention.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

AI Systems Engineer - Scalable Training Infra
AI Systems Engineer - Scalable Training Infra

OpenAI • San Francisco (CA)

On-site
USD 160,000 - 210,000
AI Training Optimization Scientist: Scale Efficiently
AI Training Optimization Scientist: Scale Efficiently

United States Digital Space LLC • United States

Remote
USD 140,000 - 230,000
Senior AI Infrastructure Engineer - Edge & API Systems
Senior AI Infrastructure Engineer - Edge & API Systems

United States Digital Space LLC • United States

Remote
USD 194,000 - 266,000
Senior AI Engineer, Platform Infra & CI/CD
Senior AI Engineer, Platform Infra & CI/CD

Spacex • Redmond (WA)

On-site
USD 220,000 - 350,000
Stock options
Long-term incentives
401(k) plan
+5
AI Enablement Engineer — Scale AI Across Engineering
AI Enablement Engineer — Scale AI Across Engineering

United States Digital Space LLC • United States

Remote
USD 120,000 - 180,000
Distributed AI Training Infrastructure Engineer
Distributed AI Training Infrastructure Engineer

Fireworks AI • United States

Remote
USD 130,000 - 210,000
GenAI Infra Engineer — Build Scalable AI Platform
GenAI Infra Engineer — Build Scalable AI Platform

United States Digital Space LLC • United States

Remote
USD 110,000 - 170,000
Senior AI Training Systems Architect
Senior AI Training Systems Architect

Designworks Talent LLC • Bellevue (WA)

Hybrid
USD 180,000 - 260,000
Medical Insurance
Dental Insurance
Vision Insurance
+2
AI Training Infrastructure Engineer – Distributed GPU
AI Training Infrastructure Engineer – Distributed GPU

FDE Times • San Mateo (CA)

On-site
USD 140,000 - 230,000
AI Engineer: Build, Evaluate & Govern Intelligent Systems
AI Engineer: Build, Evaluate & Govern Intelligent Systems

Digital Edify • United States

On-site
USD 120,000 - 180,000