Senior Staff Engineer - AI Data Path

Posted Yesterday
Hiring Remotely in California, USA
Remote or Hybrid
Senior level
Artificial Intelligence • Machine Learning • Software • Analytics
The Role
Leads hands-on design, development, and optimization of AI data movement and distributed storage systems. Responsibilities include integrating NVIDIA NIXL and DDN Infinia with GPU inference platforms, optimizing GPU-to-storage I/O using GPUDirect Storage, RDMA, and NVMe-over-Fabrics, developing KV cache and multi-tier storage strategies, benchmarking production systems, resolving performance bottlenecks, influencing distributed inference architecture, and mentoring engineers.
Summary Generated by Built In

DDN is seeking a highly experienced Senior Staff Engineer specializing in AI Data Path & Storage to lead hands-on development and integration of advanced storage systems with next-generation AI inference pipelines. This role involves coding, prototyping, and rapidly iterating on solutions in close collaboration with architects to design and deliver high-performance data movement architectures. You will leverage NVIDIA’s NIXL (Inference Transfer Library) alongside the Infinia Data Intelligence Platform to enable ultra-low-latency, high-throughput data movement across GPU, memory, and distributed storage layers, including workloads involving KV cache management and vector database retrieval. The ideal candidate brings deep expertise in distributed storage, GPU data paths, and large-scale system optimization, with a proven track record of building and shipping production-grade AI infrastructure.

 
Key Responsibilities
  • Lead the design and implementation of high-performance data movement pipelines using NVIDIA NIXL across GPU, CPU, and storage tiers.

  • Architect and drive integration of DDN Infinia with GPU-accelerated inference platforms for large-scale, real-time AI workloads.

  • Own end-to-end optimization of I/O paths between GPU memory and storage using technologies such as NVIDIA GPUDirect Storage, RDMA, and NVMe-over-Fabrics.

  • Define and implement multi-tier storage architectures (NVMe, SSD, object storage) optimized for inference latency, throughput, and scalability.

  • Lead development of advanced KV cache management strategies, including offloading, prefetching, and persistence across distributed storage layers.

  • Partner with AI/ML engineering teams to optimize inference performance in frameworks such as PyTorch and TensorFlow.

  • Establish benchmarking frameworks and lead performance tuning efforts for storage and data movement in production inference environments.

  • Diagnose and resolve complex system bottlenecks across storage, networking, and GPU subsystems.

  • Influence architecture decisions for distributed inference systems, ensuring scalability, resilience, and efficient data locality.

  • Drive engineering excellence through best practices in observability, performance monitoring, automation, and reliability engineering.

  • Mentor junior engineers and provide technical leadership across cross-functional teams.

 
Required Qualifications
  • Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field.

  • 12+ years of experience in storage systems, distributed systems, or performance engineering.

  • Proven track record of architecting and delivering large-scale, high-performance infrastructure systems.

  • Deep expertise in distributed storage architectures (object storage, scalable file systems, or cloud-native storage platforms).

  • Strong understanding of Linux I/O stack, filesystem internals, and storage protocols.

  • Extensive hands-on experience with NVMe, SSD optimization, and high-performance storage environments.

  • Strong experience with RDMA, InfiniBand, or other high-speed data transfer technologies.

  • Solid understanding of GPU computing concepts and CPU–GPU data movement patterns.

  • Proficiency in Python and/or C/C++, with advanced debugging, profiling, and performance tuning skills.

  • Demonstrated ability to optimize latency-sensitive, high-throughput production systems.

Preferred Skills
  • Hands-on experience with NVIDIA NIXL or similar data movement frameworks.

  • Experience with GPU-aware storage pipelines and GPUDirect Storage.

  • Strong understanding of AI inference systems, LLM serving architectures, and KV cache optimization.

  • Experience with Retrieval-Augmented Generation (RAG) pipelines and open vector search ecosystems.

  • Background in high-performance computing (HPC) or hyperscale distributed environments.

  • Expertise in caching strategies, memory tiering, and data locality optimization.

  • Experience designing disaggregated compute and storage architectures.

 
What You’ll Work On
  • Leading the evolution of storage systems into GPU-native data layers for AI inference

  • Building next-generation distributed AI infrastructure using NIXL and Infinia

  • Driving performance breakthroughs in real-time LLM inference at scale

  • Designing storage architectures for large-scale AI datasets and retrieval systems

Skills Required

  • Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field
  • 12+ years of experience in storage systems, distributed systems, or performance engineering
  • Experience architecting and delivering large-scale, high-performance infrastructure systems
  • Deep expertise in distributed storage architectures, including object storage, scalable file systems, or cloud-native storage platforms
  • Strong understanding of the Linux I/O stack, filesystem internals, and storage protocols
  • Hands-on experience with NVMe, SSD optimization, and high-performance storage environments
  • Experience with RDMA, InfiniBand, or other high-speed data transfer technologies
  • Understanding of GPU computing concepts and CPU-GPU data movement patterns
  • Proficiency in Python and/or C/C++
  • Advanced debugging, profiling, and performance tuning skills
  • Demonstrated ability to optimize latency-sensitive, high-throughput production systems
  • Experience with NVIDIA NIXL or similar data movement frameworks
  • Experience with GPU-aware storage pipelines and GPUDirect Storage
  • Understanding of AI inference systems, LLM serving architectures, and KV cache optimization
  • Experience with Retrieval-Augmented Generation pipelines and open vector search ecosystems
  • Background in high-performance computing or hyperscale distributed environments
  • Expertise in caching strategies, memory tiering, and data locality optimization
  • Experience designing disaggregated compute and storage architectures
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: Chatsworth, CA
706 Employees
Year Founded: 1998

What We Do

DDN is the world’s largest private data storage company and the leading provider of intelligent technology and infrastructure solutions for Enterprise At Scale, AI and analytics, HPC, government and academia customers. Through its DDN and Tintri divisions, the company delivers AI, Data Management software and hardware solutions, and unified analytics frameworks to solve complex business challenges for data-intensive, global organizations. DDN provides its enterprise customers with the most flexible, efficient and reliable data storage solutions for on-premises and multi-cloud environments at any scale. Over the last two decades, DDN has established itself as the data management provider of choice for over 11,000 enterprises, government, and public-sector customers, including many of the world’s leading financial services firms, life science organizations, manufacturing and energy companies, research facilities, and web and cloud service providers.

Similar Jobs

Grow Therapy Logo Grow Therapy

Provider Growth Associate

Healthtech • Social Impact • Software
Remote or Hybrid
USA
650 Employees
31-31 Hourly

Gusto Logo Gusto

Staff Software Engineer

Fintech • HR Tech
Easy Apply
Remote or Hybrid
United States
4405 Employees
163K-247K Annually

Toast Logo Toast

Business Development Representative

Cloud • Fintech • Food • Information Technology • Software • Hospitality
Remote
US
5000 Employees
75K-120K Annually

Hewlett Packard Enterprise Logo Hewlett Packard Enterprise

Storage Channel Specialist

Artificial Intelligence • Cloud • Information Technology • Consulting
In-Office or Remote
2 Locations
85422 Employees
126K-253K Annually

Similar Companies Hiring

Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
60 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
30 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
43 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account