Staff Engineer, Distributed Storage and HPC & AI Infrastructure
Together Ai
San Francisco, CA · Posted June 4, 2026
Just laid off? Start here: a plan for the first weeks
See every open role at Together Ai
More in California job market · Technology & Software · Data & Analytics jobs
More roles at Together Ai
- Product Manager, AI Infrastructure · San Francisco, CA
- Data Center Operations Coordinator · San Francisco, CA
- Forward Deployed Engineer, Post-Training · San Francisco, CA
- Strategic Finance Senior Associate - Compute · San Francisco, CA
- Product Marketing Manager, Partners · San Francisco, CA
- Senior Technical Recruiter, AI/ML Research · San Francisco, CA
Similar jobs
- Principal Data Scientist, Economy Recommendations · Roblox · San Mateo, CA
- Senior Manager, Data Analytics · Life360 · San Mateo, CA
- Principal Data Scientist · Autodesk · CA
- Senior Manager, Business Analytics · Autodesk · CA
- Associate Director, Regulatory Data Quality · Gilead Sciences · Foster City, CA
- Senior Manager, Safety AI · Samsara · CA
Where San Francisco, CA ranks
Where San Francisco, CA placed in our monthly rankings, month by month. Each month stays online at its own address.
November 2026
- #1Best cities for security jobs316 openings · of 20
- #1Best cities for technology jobs7,178 openings · of 100
- #1Best cities for financial services pay after cost of living$173k · of 33
- #1Best cities for entry-level jobs that want AI skills35% · of 100
- #1Best cities for entry-level jobs that want AI skills in the West Coast35% · of 54
- #1Cities with the most jobs that do not need a four-year degree in the West Coast902 jobs · of 47
Rankings about data & analytics jobs
- Best cities for data and analytics jobs #1 New York (1,713 openings)
- Best cities for data and analytics pay after cost of living #1 Foster ($232k)
- Most requested AI skills in data and analytics jobs #1 Machine learning engineering (17.1%)
- AI tools named most often in data and analytics jobs #1 Databricks (7.6%)
- Employers with the most data and analytics jobs #1 Amazon (711 roles)
Data & Analytics jobs · Top Data & Analytics employers · All rankings
Skills this job asks for
Say which you have and which you want to build. We use it to match you to jobs like this one.
About the job
About the Role
In this role, you will operate, scale, and optimize multi-petabyte storage systems purpose-built for the world’s largest AI training and inference workloads. Storage now sits on the inference critical path as well, so you’ll also build the caching and data-movement layers that deliver KV cache and model weights directly into GPU memory at fabric speed. You’ll manage and scale high-performance parallel filesystems and object stores, evaluate and integrate cutting-edge technologies such as Vast, Weka, Ceph, and Lustre, and solve the complex engineering challenges of operating at extreme throughput, low-latency data paths, and massive cluster-scale storage operations.
You will also build Kubernetes-native storage operators and self-service platforms that provide automated provisioning, strict multi-tenancy, performance isolation, and quota enforcement at cluster scale. Day-to-day, you’ll optimize end-to-end data paths for 10-50 GB/s per node, design multi-tier caching architectures, implement intelligent prefetching and model-weight distribution, build KV-cache offload and reuse tiers for inference, and tune parallel filesystems for AI workloads.
Responsibilities
Architect and implement the technical strategy and storage roadmap for Together AI, driving high-performance architectural decisions as we scale our GPU fleet.
Engineer and scale multi-petabyte AI/ML storage systems by integrating Vast, Weka, and Ceph while executing deep cost optimization through automated tiering and lifecycle policies.
Develop intelligent caching and tiered storage architectures to achieve extreme IOPS and cluster-wide throughput at GPU scale for training and inference workloads, including GPU-direct (RDMA) delivery of model weights for fast cold starts and autoscaling.
Design and build KV-cache offload and reuse for inference: tiering KV blocks across GPU HBM, host memory, local NVMe, and distributed cache, enabling prefix reuse across requests and nodes, and integrating with inference engines (vLLM, SGLang, TensorRT-LLM) to cut time-to-first-token.
Tune storage isolation at the L2/L3 network layers to ensure secure, production-grade multi-tenancy for storage clients.
Code Kubernetes storage operators and controllers to enable automated provisioning, self-service abstractions, and quota enforcement.
Engineer end-to-end data paths to achieve 10+ GB/s per GPU node; architect multi-tier caching for model weights, KV cache, and datasets; tune parallel filesystems using advanced profiling; and scale storage infrastructure across thousands of nodes.
Optimize end-to-end data paths through advanced benchmarking and profiling, contributing high-impact code to open-source storage and inference projects and internal tooling.
Requirements
8+ years in storage engineering, managing distributed storage at multi-petabyte scale
Proven track record building, deploying and operating high-performance storage for GPU/HPC clusters, for training and/or inference
Experience scaling storage for inference: KV-cache offload or disaggregated KV storage, prefix caching, model-weight distribution, or similar latency-critical data paths
Deep Kubernetes and cloud-native storage experience in production environments
Strong coding skills in Go and Python with demonstrated ability to build production-grade systems and tooling
BS/MS in Computer Science, Engineering, or equivalent practical experience
History of technical leadership: designing systems that significantly improved performance, reliability (99.999%+ uptime), or cost efficiency
Distributed Storage Systems: Deep expertise in either of Ceph, WekaFS, Lustre, Vast, GPFS, or similar parallel filesystems at multi-petabyte scale
Object Storage: Production experience with S3, MinIO, Ceph, or R2 including performance optimization and cost management
Kubernetes Storage: CSI drivers, StatefulSets, PersistentVolumes, storage operators, and custom controllers
Storage optimization for GPU workloads, RDMA/InfiniBand networking, parallel filesystem optimization (TB/s aggregate cluster throughput - line saturation)
Programming: Go and Python for automation, operators, and tooling; C/C++ or Rust for performance-critical data paths
Infrastructure as Code: Terraform, Ansible, Helm, GitOps (ArgoCD)
Linux Storage Stack: Advanced knowledge of filesystems (ext4, xfs), LVM, NVMe optimization, RAID configurations
Observability: Prometheus, Grafana, Thanos architecture and operations
Nice to Have
GPU Direct Storage (GDS), NVMe-oF, storage networking, RDMA implementations
GPUDirect RDMA and zero-copy data movement (ibverbs, NIXL, UCX), and KV-cache transfer frameworks (e.g. LMCache, Mooncake, Dynamo)
ML/AI storage patterns (model weights, checkpointing, dataset caching, KV-cache offload and prefix caching)
Storage benchmarking and profiling tools (fio, iperf3, iostat, blktrace)
Patents, publications (FAST, OSDI, SOSP, SC, ATC), or significant open-source contributions in storage or systems
About Together AI
Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society, and together we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI, and our team has been behind technological advancement such as FlashAttention, Hyena, FlexGen, and RedPajama. We invite you to join a passionate group of researchers in our journey in building the next generation AI infrastructure.
Compensation
We offer competitive compensation, startup equity, health insurance, and other benefits, as well as flexibility in terms of remote work. The US base salary range for this full-time position is: $250,000 - $330,000 + equity + benefits. Our salary ranges are determined by location, level and role. Individual compensation will be determined by experience, skills, and job-related knowledge.
Application Policy
Please read each role carefully and apply to the position best suited to you. Candidates are limited to 6 applications within a 90-day period, and applications beyond that limit will be automatically rejected. You may apply to additional roles 90 days after your last application.
Equal Opportunity
Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.
Please see our privacy policy at https://www.together.ai/privacy