Staff Engineer, Distributed Storage and HPC & AI Infrastructure

Posted 5 Days Ago
Be an Early Applicant
Hiring Remotely in Bangalore, Bengaluru Urban, Karnataka, IND
In-Office or Remote
Senior level
Artificial Intelligence • Information Technology
The Role
Operate and scale multi-petabyte distributed storage for AI and HPC workloads. Design high-performance parallel filesystems, object stores, caching architectures, and Kubernetes-native storage operators. Optimize data paths, throughput, multi-tenancy, cost, and reliability across thousands of GPU nodes. Integrate technologies such as Vast, Weka, Ceph, and Lustre; develop automation and tooling in Go and Python; and contribute to storage infrastructure, observability, benchmarking, and open-source projects.
Summary Generated by Built In
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. 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, 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.
  • 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 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 projects and internal tooling.
Requirements
  • 8+ years in storage engineering, managing distributed storage at multi-petabyte scale
  • Proven track record deploying and operating 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 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
  • 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 Skills
  • GPU Direct Storage (GDS), NVMe-oF, storage networking, RDMA implementations
  • ML/AI storage patterns (model weights, checkpointing, dataset caching)
  • Storage benchmarking and profiling tools (fio, iperf3, iostat, blktrace)

About Together AI

Together AI, the AI Native Cloud, is purpose-built for AI engineers. AI application developers get high-performance inference that scales reliably, fine-tuning and reinforcement learning for creating frontier-level specialized models, and pre-training at massive scale for fully custom intelligence, all around a marketplace of leading open models that teams can run, adapt, and own. Trusted by Cursor, Decagon, ElevenLabs, Salesforce, and Zoom, Together serves 400+ trillion tokens a month.

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 


Skills Required

  • 8+ years of storage engineering experience managing distributed storage at multi-petabyte scale
  • Experience deploying and operating high-performance storage for GPU or HPC clusters
  • Deep Kubernetes and cloud-native storage experience in production environments
  • Strong Go and Python coding skills for production-grade systems and tooling
  • BS or MS in Computer Science, Engineering, or equivalent practical experience
  • Technical leadership designing systems that improve performance, reliability, or cost efficiency
  • Deep expertise in Ceph, WekaFS, Lustre, Vast, GPFS, or similar parallel filesystems at multi-petabyte scale
  • Production experience with S3, MinIO, Ceph, or R2, including performance optimization and cost management
  • Experience with Kubernetes storage technologies including CSI drivers, StatefulSets, PersistentVolumes, storage operators, and custom controllers
  • Experience optimizing storage for GPU workloads, RDMA or InfiniBand networking, and parallel filesystems
  • Experience with Terraform, Ansible, Helm, and GitOps such as ArgoCD
  • Advanced knowledge of Linux storage, ext4, XFS, LVM, NVMe optimization, and RAID
  • Experience operating observability platforms including Prometheus, Grafana, and Thanos
  • Experience with GPU Direct Storage, NVMe-oF, storage networking, or RDMA implementations
  • Experience with ML/AI storage patterns including model weights, checkpointing, and dataset caching
  • Experience with storage benchmarking and profiling tools such as fio, iperf3, iostat, and blktrace
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Surry Hills
84 Employees
Year Founded: 2022

What We Do

Together AI is a research-driven artificial intelligence company. We contribute leading open-source research, models, and datasets to advance the frontier of AI. Our decentralized cloud services empower developers and researchers at organizations of all sizes to train, fine-tune, and deploy generative AI models. We believe open and transparent AI systems will drive innovation and create the best outcomes for society

Similar Jobs

Capco Logo Capco

IT Control Testing

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
India
6000 Employees

Atlassian Logo Atlassian

Senior Engineering Manager

Cloud • Information Technology • Productivity • Security • Software • App development • Automation
In-Office or Remote
Bengaluru, Bengaluru Urban, Karnataka, IND
11000 Employees

Atlassian Logo Atlassian

Program Manager

Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Remote
India
11000 Employees

The Aerospace Corporation Logo The Aerospace Corporation

Integrity &Compl Specialist II

Aerospace • Artificial Intelligence • Cloud • Machine Learning • Software • Cybersecurity • Defense
Remote or Hybrid
India
4600 Employees

Similar Companies Hiring

Kepler  Thumbnail
Artificial Intelligence • Fintech • Software
New York, New York
9 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
30 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account