Staff Engineer, Distributed Storage and Hpc & AI Infrastructure

Together AI Together AI · Data AI · San Francisco, CA · Engineering

Staff Engineer focused on designing and delivering multi-petabyte storage systems optimized for AI training and inference workloads. Responsibilities include architecting high-performance parallel filesystems and object stores, building Kubernetes-native storage operators, optimizing data paths for high throughput, and implementing intelligent caching and data distribution strategies. The role requires deep expertise in distributed storage systems, Kubernetes, and programming in Go and Python.

What you'd actually do

  1. Design multi-petabyte AI/ML storage systems; integrate WekaFS, Ceph, etc.; lead capacity planning and cost optimization (30-50% savings via tiering, lifecycle policies, right-sizing).
  2. Design/optimize RDMA, InfiniBand, 400GbE networks; tune for max throughput/min latency; implement NVMe-oF/iSCSI; troubleshoot bottlenecks; optimize TCP/IP for storage.
  3. Build Kubernetes storage operators/controllers; enable automated provisioning, self-service abstractions, multi-tenant isolation, quotas; create reusable Helm/Terraform patterns.
  4. Deliver 10-50 GB/s per GPU node; optimize caching (weights/datasets/checkpoints), parallel filesystems, and data paths; troubleshoot with profiling tools; scale to thousands of nodes.
  5. Build multi-tier caches (local NVMe, distributed, object); optimize data locality and model-weight distribution; implement smart prefetching/eviction.

Skills

Required

  • Distributed storage systems (WekaFS, Ceph, Lustre, GPFS, BeeGFS)
  • Object storage (S3, MinIO, Ceph, R2)
  • Kubernetes storage (CSI drivers, StatefulSets, PersistentVolumes, operators)
  • Go
  • Python
  • RDMA
  • InfiniBand
  • 400GbE networking
  • NVMe-oF/iSCSI
  • TCP/IP optimization
  • Helm
  • Terraform
  • Prometheus
  • Grafana
  • Thanos

Nice to have

  • GPU Direct Storage (GDS)
  • 100GbE/400GbE storage networking
  • ML/AI storage patterns (model weights, checkpointing, dataset caching)
  • Kubernetes operator development (controller-runtime, kubebuilder)
  • Storage snapshots, cloning, and thin provisioning
  • Backup and disaster recovery (Velero, Restic, cross-region replication)
  • Storage encryption
  • Storage benchmarking and profiling tools (fio, iperf3, iostat, blktrace)

What the JD emphasized

  • multi-petabyte scale
  • high-performance storage for GPU/HPC clusters
  • Deep Kubernetes and cloud-native storage experience in production environments
  • Strong coding skills in Go and Python with demonstrated ability to build production-grade tools
  • Distributed Storage Systems: Deep expertise in WekaFS, Lustre, GPFS, BeeGFS, or similar parallel filesystems at multi-petabyte scale
  • Kubernetes Storage: CSI drivers, StatefulSets, PersistentVolumes, storage operators, and custom controllers
  • Storage optimization for GPU workloads, RDMA/InfiniBand networking, parallel filesystem optimization (100+ GB/s aggregate cluster throughput)
  • Observability: Prometheus, Grafana, Thanos architecture and operations

Other signals

  • design and deliver multi-petabyte storage systems purpose-built for the world’s largest AI training and inference workloads
  • architect high-performance parallel filesystems and object stores
  • build Kubernetes-native storage operators and self-service platforms
  • optimize end-to-end data paths for 10-50 GB/s per node
  • implement intelligent prefetching and model-weight distribution