Data Direct Networks logo

Senior Staff Engineer - AI Data Path

Data Direct Networks·Salary not specified

Primary stack

PythonC++

Nice to have's

Infinia Data Intelligence Platformvector database retrievalKV cache managementPyTorchTensorFlowRetrieval-Augmented Generation (RAG)high-performance computing (HPC)

Job description

About the Position

Senior Staff Engineer - AI Data Path

Santa Clara Office

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.

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.

Requirements

  • 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.

Nice to Have

  • 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.

© Data Direct Networks. This job description was sourced from the employer's public career page. TheJob is not the employer — we index the posting and route candidates to the source. All content rights and hiring decisions belong to the employer.

DataDirect Networks (DDN.com) is a world leader in massively-scalable storage platforms and solutions engineered for the Big Data and Cloud era. Our storage, processing solutions and professional services enable content-rich and high-data growth environments to achieve the highest levels of systems performance, scalability and efficiency.In solving today’s Big Data lifecycle management challenges at scale, DDN enables organizations to maximize the value of their information, everywhere. As the technology behind the world’s leading data-driven organizations, our customers include online content and social networking providers, high performance cloud and grid computing, life sciences, financial services, media production houses and security and intelligence agencies.

More at Data Direct Networks

All 95 roles

Popular searches