aap3 logo

Sr AI Infrastructure Engineer

aap3
  • Costa Mesa, CA
  • Full-time
  • Instant Apply
2 days ago

Job Description

Senior AI Infrastructure Engineer to lead the vision, execution, and long-term stability of how Company trains with GPUs at scale. In this role, you will take absolute ownership of cluster robustness, ensuring our high-performance GPU systems are highly available, fault-tolerant, and resilient for ML platform and research teams company-wide. This is a highly hands-on role where your primary focus is logical stability and automated resilience—building self-healing mechanisms to proactively detect and isolate hardware faults, tuning NCCL and high-speed networking, and optimizing Kubernetes, Run:AI, and Ray scheduling. By replacing manual triage with automated deployment tooling and deep observability, you will ensure our massive-scale training infrastructure runs seamlessly and scales without linear headcount growth.

Job Responsibilities

Rack, stack, cable, and bring up GPU compute (H200/B200/B300, NVL72) including physical topology, power, cooling, firmware/BIOS, and burn in validation.
Build and tune the interconnect fabric (NVLink, InfiniBand, RoCE, Spectrum-X) connecting hundreds of GPUs into low latency training and inference clusters.
Integrate high performance parallel storage (VAST, DDN, Weka) to sustain the throughput demanded by distributed training and terabyte scale multi modal datasets across Company's programs.
Automate cluster deployment and configuration end to end, including infrastructure as code for bring up, firmware/driver management, and fabric config, so new capacity comes online with minimal manual work.
Operate and extend our Kubernetes/Run:AI environment for GPU scheduling, quota management, and multi tenant workload isolation across research and engineering teams company wide.
Own fleet health: monitoring, alerting, and rapid triage of hardware and network faults (bad transceivers, GPU Xid errors, NCCL/collective failures, RoCE congestion).
Onboard engineers and researchers onto the platform and act as their escalation point, working directly alongside them to debug, train, and optimize their workloads whenever infrastructure, not the model, is the bottleneck.
Partner with product facing teams across Company to understand emerging compute needs and translate them into platform capability.

Numbers & Facts

LocationCosta Mesa, CA
Job TypeFull-time

Qualifications

10+ years in a hands on infrastructure, HPC, or datacenter engineering role supporting GPU compute at scale.
Hands on experience with H200/B200/B300 (or comparable) GPU systems: bring up, cabling, firmware/driver management.
Experience with high performance interconnects (NVLink, InfiniBand, RoCE, Spectrum-X) in clusters of hundreds of GPUs.
Experience with high performance parallel storage (VAST, DDN, Weka, Lustre, or similar).
Kubernetes required; Run:ai or similar GPU scheduling/orchestration experience strongly preferred.
Strong automation background. You build repeatable, automated deployment pipelines rather than manual processes.
Able to lift/move 50+ lbs and perform physical datacenter work (rack/stack/cable/troubleshoot).
Eligible to obtain and maintain an active U.S. Top Secret clearance.
PREFERRED QUALIFICATIONS

Experience with NVIDIA NVL72 rack scale systems.
Experience supporting LLM token serving/inference infrastructure alongside training clusters.
Network fabric tuning experience (congestion control, adaptive routing, QoS) for RoCE/InfiniBand at scale.
Familiarity with GPU/network observability tooling (DCGM, fabric telemetry) and automated fault detection.
Experience supporting infrastructure as a shared platform serving multiple internal customer teams with differing requirements.

About Company

Company is a defense technology company with a mission to transform U.S. and allied military capabilities with advanced technology. By bringing the expertise, technology, and business model of the 21st century’s most innovative companies to the defense industry, Company is changing how military systems are designed, built and sold. Company’s family of systems is powered by Lattice OS, an AI-powered operating system that turns thousands of data streams into a realtime, 3D command and control center. As the world enters an era of strategic competition, Company is committed to bringing cutting-edge autonomy, AI, computer vision, sensor fusion, and networking technology to the military in months, not years.

Skills

  • Artificial Intelligence (AI)unmatched
  • Broadbandunmatched
  • GPU (Graphics Processing Unit)unmatched
  • High Availabilityunmatched
  • Home Automationunmatched
  • Machine Toolunmatched
  • Performance Managementunmatched

Be found by employers

5,500+ employers search our resume database daily. Add yours to get found by recruiters looking for candidates like you.

Level up your application

Professional resume templates

Browse dozens of recruiter approved resume templates, layouts and formats. Choose your favorite and make it your own in minutes.

Free resume templates

Free resume builder

Improve your existing resume or start from scratch and create a standout, ATS-friendly resume. Add job-specific content, download and apply.

Free resume builder