Member of Technical Staff (Software Engineer, Inference & Training Platform)

United States Digital Space LLC

San Francisco (CA)

Hybrid

USD 180,000 - 240,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Job summary

United States Digital Space LLC is seeking an infrastructure leader to own a self-serve GPU compute platform for training and inference workloads. You will design and operate the system that lets researchers launch jobs across multi-cloud GPU fleets without manual provisioning.

You will own provisioning, scheduling, and reliability across providers, building fault-tolerance, observability, and a coherent platform roadmap for scalable AI workloads.

Qualifications

  • Deep Kubernetes experience with custom operators and CRDs.
  • Experience managing GPU clusters at scale with NVIDIA GPUs and high-speed networks.
  • Experience orchestrating compute across multiple clouds.
  • Strong distributed systems fundamentals.
  • Proficiency in Go, Rust or C++ for infrastructure.
  • Experience with long-running training and high-availability inference.
  • Ability to own problems end-to-end.

Responsibilities

  • Build a self-serve compute platform for training and inference workloads.
  • Operate the GPU fleet across providers with provisioning and lifecycle management.
  • Develop scheduling and placement to balance capacity across clouds.
  • Support both long-running training jobs and production inference workloads.
  • Own Kubernetes orchestration across multiple clusters and providers.
  • Implement fault tolerance, autoscaling, and observability.
  • Set technical direction across teams to align platform roadmap.

Skills

Kubernetes
GPU clusters
Multi-cloud orchestration
Distributed systems
Go/Rust/C++

Tools

Custom Kubernetes Operators

Job description

the company serves hundreds of millions of queries a month, and every one of them fans out into multiple AI inference requests running in real time. Behind that sits a large GPU fleet spread across several cloud providers. Today, our inference engineers and researchers build models while also managing networking, securing capacity, and operating the underlying GPU clusters, responsibilities we want a dedicated platform team to own. Your job is to take ownership of that infrastructure and hide its complexity behind a unified, self-serve platform for running training and inference workloads.

Responsibilities
  • Build a self-serve compute platform. Design and own the systems that let inference engineers and researchers launch training jobs and operate inference services without managing GPU provisioning, cluster configuration, or provider-specific infrastructure.
  • Operate the GPU fleet. Own provisioning, lifecycle management, reliability, and capacity integration across providers, giving teams a consistent way to use compute regardless of where it runs.
  • Solve for GPU scarcity. Build the scheduling and placement logic that finds available capacity across providers, packs it efficiently, and gets the right workload onto the right hardware under real constraints.
  • Support two very different workloads. Keep long-running distributed training jobs healthy while simultaneously guaranteeing the availability and latency of production inference services on the same fleet.
  • Own the Kubernetes for GPU orchestration. Write the operators and CRDs, and manage many clusters across providers so the platform behaves the same everywhere we run.
  • Make failure boring. Build the fault tolerance, autoscaling, and observability that keep the fleet utilized and let workloads survive node loss, provider hiccups, and capacity shifts without human intervention.
  • Set technical direction across teams. Partner with inference and cloud infrastructure engineers to turn operational constraints into a coherent platform architecture and roadmap.
Qualifications
  • Deep Kubernetes experience — custom operators, CRDs, and multi-cluster federation, not just running kubectl apply.
  • You've managed GPU clusters at scale: NVIDIA hardware, CUDA, and the networking that makes them fast (InfiniBand or RoCE).
  • You've orchestrated compute across multiple clouds (CoreWeave, AWS, GCP, or similar) and understand how different each one really is.
  • Strong distributed systems fundamentals: scheduling, resource allocation, and fault tolerance under load.
  • You write infrastructure and systems-level code in Go, Rust or C++.
  • You've supported both long-running training jobs and high-availability inference services, and you know why they pull infrastructure in opposite directions.
  • You own problems end-to-end and do well when the path forward isn't laid out for you.
Additional experience we value
  • Inference serving stacks: vLLM, SGLang, or TensorRT-LLM.
  • Slurm or other HPC schedulers.
  • GPU kernel work in CUDA or Triton — not required, but notable.
  • High-speed interconnects: InfiniBand, RoCE, or RDMA in production.
  • Observability for ML workloads: Prometheus, Grafana, or Weights & Biases.

If you’re excited about this role, we encourage you to apply even if your experience doesn’t match every qualification listed above.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Member of Technical Staff – Software Engineer, GPU Cluster Infrastructure
Member of Technical Staff – Software Engineer, GPU Cluster Infrastructure

Perplexity • San Francisco (CA)

On-site
USD 180,000 - 240,000
Member of Technical Staff (Software Engineer, GPU Cluster Infrastructure)
Member of Technical Staff (Software Engineer, GPU Cluster Infrastructure)

United States Digital Space LLC • San Francisco (CA)

On-site
USD 180,000 - 260,000
Member of Technical Staff (Software Engineer, Inference & Training Platform) Perplexity AI San [...]
Member of Technical Staff (Software Engineer, Inference & Training Platform) Perplexity AI San [...]

Neura Market • San Francisco (CA)

On-site
USD 180,000 - 260,000
Member of Technical Staff (Software Engineer, GPU Cluster Infrastructure)
Member of Technical Staff (Software Engineer, GPU Cluster Infrastructure)

B Capital • United States

On-site
USD 180,000 - 230,000
Member of Technical Staff - GPU Infrastructure
Member of Technical Staff - GPU Infrastructure

Prime Intellect • United States

On-site
USD 120,000 - 150,000
Cluster Engineer
Cluster Engineer

STN Inc • San Francisco (CA)

On-site
USD 180,000 - 240,000
Senior Inference Platform Engineer - Data Center
Senior Inference Platform Engineer - Data Center

Hamilton Barnes Associates Limited • San Francisco (CA)

On-site
USD 270,000 - 330,000
Equity
Member of Technical Staff — Compute Cluster
Member of Technical Staff — Compute Cluster

Kindredventures • San Francisco (CA)

On-site
USD 140,000 - 230,000
Software Engineer, AI Infrastructure
Software Engineer, AI Infrastructure

Harell Data • Palo Alto (CA)

On-site
USD 180,000 - 260,000
Member of Technical Staff — Compute Cluster
Member of Technical Staff — Compute Cluster

Causal • San Francisco (CA)

On-site
USD 180,000 - 240,000