Staff Engineer: GPU-Scale Distributed Storage & AI Infra

Fdetimes

San Francisco (CA)

Remote

USD 250,000 - 330,000

Full time

14 days+

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Startup equity
Health insurance
Remote work flexibility
Flexible benefits

Job summary

Together AI seeks an experienced storage engineer to design and operate scalable AI storage infrastructure for multi-petabyte workloads. You will optimize end-to-end data paths, build caching tiers, and integrate advanced storage technologies to support training and inference at scale.

You will architect GPU-focused storage architectures, implement KV-cache offload, and drive multi-tenant isolation while contributing to open-source projects and internal tooling.

Qualifications

  • 8+ years in storage engineering at multi-petabyte scale.
  • Proven track record building and operating high-performance storage for GPU/HPC.
  • Experience scaling for inference: KV-cache offload or similar.
  • Strong Go and Python coding for production-grade systems.

Responsibilities

  • Architect and implement storage roadmap for Together AI at scale.
  • Engineer multi-petabyte AI/ML storage with Vast, Weka, Ceph, and Lustre.
  • Develop caching and tiered storage for extreme IOPS and throughput.
  • Build KV-cache offload across GPU memory, NVMe, and distributed cache.
  • Tune isolation for multi-tenant storage in production.
  • Code Kubernetes storage operators and quota enforcement.

Skills

Go
Python
Kubernetes
Distributed storage
GPU/HPC
Performance optimization

Education

BS/MS in CS/Engineering

Tools

Ceph
WekaFS
Lustre
Vast
GPFS
S3/MinIO

Job description

Together AI seeks an experienced storage engineer to design and operate scalable AI storage infrastructure for multi-petabyte workloads. You will optimize end-to-end data paths, build caching tiers, and integrate advanced storage technologies to support training and inference at scale.

You will architect GPU-focused storage architectures, implement KV-cache offload, and drive multi-tenant isolation while contributing to open-source projects and internal tooling.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Senior Storage Engineer -AI Infrastructure
Senior Storage Engineer -AI Infrastructure

ARC Cloud • San Jose (CA)

On-site
USD 150,000 - 230,000
Storage Engineer (AI Infrastructure) - Hosting
Storage Engineer (AI Infrastructure) - Hosting

Hamilton Barnes • San Francisco (CA)

Remote
USD 170,000 - 230,000
Stock options
Remote work allowance
Lead Storage Software Engineer for AI-Scale HPC
Lead Storage Software Engineer for AI-Scale HPC

NVIDIA Corporation • Santa Clara (CA)

Hybrid
USD 224,000 - 431,000
Equity
Comprehensive Benefits
AI Storage Systems Architect for GPU-Cloud AI
AI Storage Systems Architect for GPU-Cloud AI

Bitdeer • San Jose (CA)

Hybrid
USD 140,000 - 210,000
Senior Storage Engineer - AI Infra & GPU Clusters (Remote)
Senior Storage Engineer - AI Infra & GPU Clusters (Remote)

Hamilton Barnes • San Francisco (CA)

Remote
USD 170,000 - 230,000
Stock options
Remote work allowance
AI Storage Architect for High-Performance GPU Cloud
AI Storage Architect for High-Performance GPU Cloud

Bitdeer (NASDAQ: BTDR) • Austin (TX)

On-site
USD 140,000 - 200,000
Senior Data Infrastructure Engineer (GPU/AI)
Senior Data Infrastructure Engineer (GPU/AI)

Fireworks AI • San Mateo (CA)

On-site
USD 180,000 - 260,000
Senior Data Infrastructure Engineer for GPU AI Clusters
Senior Data Infrastructure Engineer for GPU AI Clusters

Fireworks • San Mateo (CA)

On-site
USD 230,000 - 300,000
Staff Data Infrastructure Engineer – Storage for AI GPUs
Staff Data Infrastructure Engineer – Storage for AI GPUs

New York tech • San Mateo (CA)

On-site
USD 180,000 - 240,000
Senior Storage Engineer for AI Data Infrastructure
Senior Storage Engineer for AI Data Infrastructure

NVIDIA Corporation • United States

Remote
USD 152,000 - 288,000