← All roles

Software Engineer: ML Infra

Generalist AI· Technical Staff
Apply on company site
San Francisco Bay Area (San Mateo) or Boston (Somerville)$260k - $350kFull-timePosted 3mo ago

About the role

Generalist trains very large robot foundation models with on-prem GPU fleets, demanding high-performance distributed training and inference infrastructure. Responsible for GPU compute fleets, data loading/storage in distributed environments, and robot inference orchestration.

Responsibilities

  • Owning our GPU compute fleets
  • Ensure our GPUs are easy for researchers to use and maximally utilized
  • Optimizing and improving ML data loading transport and storage in highly distributed fully utilized environments.
  • Orchestration of robot inference fleets

Qualifications

  • Have managed large fleets of GPUs doing large-scale, long-term, highly distributed training runs or inference
  • Deep experience in Slurm or Kubernetes for ML workload orchestration
  • Have build high-scale ML data loaders and preparation systems
  • Deeply understand every layer of the ML hardware, storage, and networking stacks
  • Have experience in the NVidia GPU ecosystem

Skills

Required:GPU fleet managementSlurmKubernetesML data loadingML hardware/storage/network stacksNVidia GPU ecosystem
Last updated Sep 1, 2026. Interested? Apply on the company site.
Generalist AI logo

About Generalist AI

Generalist is a frontier AI research and product driven company building general intelligence for the physical world.

Generalist AI: a frontier AI research and product driven company building general intelligence for the physical world.

AIGEN-1PRODUCTSVLAsembodied AIembodied intelligencegeneral-purposegeneral-purpose AI
Company profileWebsite

More open roles at Generalist AI