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

Perplexity Perplexity · AI Frontier · San Francisco, CA · AI Research & Systems

This role focuses on building and operating a self-serve compute platform for AI inference and training workloads, managing a large GPU fleet across multiple cloud providers, and optimizing resource utilization and scheduling. The goal is to abstract away infrastructure complexity for inference engineers and researchers.

What you'd actually do

  1. 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.
  2. Own provisioning, lifecycle management, reliability, and capacity integration across providers, giving teams a consistent way to use compute regardless of where it runs.
  3. 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.
  4. Keep long-running distributed training jobs healthy while simultaneously guaranteeing the availability and latency of production inference services on the same fleet.
  5. Write the operators and CRDs, and manage many clusters across providers so the platform behaves the same everywhere we run.

Skills

Required

  • Kubernetes (custom operators, CRDs, multi-cluster federation)
  • GPU cluster management at scale (NVIDIA hardware, CUDA)
  • Multi-cloud orchestration (CoreWeave, AWS, GCP)
  • Distributed systems fundamentals (scheduling, resource allocation, fault tolerance)
  • Go, Rust, or C++
  • Experience supporting training jobs and inference services

Nice to have

  • Inference serving stacks (vLLM, SGLang, TensorRT-LLM)
  • Slurm or other HPC schedulers
  • GPU kernel work (CUDA, Triton)
  • High-speed interconnects (InfiniBand, RoCE, RDMA)
  • Observability for ML workloads (Prometheus, Grafana, Weights & Biases)

What the JD emphasized

  • 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've supported both long-running training jobs and high-availability inference services, and you know why they pull infrastructure in opposite directions.

Other signals

  • building a self-serve compute platform for AI workloads
  • operating and optimizing a large GPU fleet across multiple cloud providers
  • solving GPU scarcity through efficient scheduling and placement
  • supporting both distributed training and real-time inference services
  • owning Kubernetes orchestration for GPUs