/draftSign in

Principal Infrastructure Engineer, AI Cluster Performance & Validation

Nscale · Houston; New York; San Francisco; Seattle

Posted by Nscale on their careers site · View the original posting

Draft this job a résumé tailored to it, and the application filled in

draft's reading

Open 65 days

The posting, as Nscale wrote it

Overview

As a Principal Infrastructure Engineer, AI Cluster Performance & Validation, you will be a critical member of the AI Infrastructure Operations team, responsible for ensuring the acceptance, performance, and scalability of our cutting-edge AI and High-Performance Computing (HPC) environments. Leveraging software engineering and testing principles, you will focus on building and maintaining the control plane, tooling, and automation that supports performance and validation testing of large-scale AI clusters. Your work will directly translate into higher system availability, compute optimization and reduced operational costs.

Key Responsibilities

  • Own the technical definition of "healthy at scale." Set the architecture, roadmap, and acceptance criteria by which multi-thousand-GPU clusters are declared production-ready, and establish the performance bar (collective bandwidth, job goodput, model FLOPs utilization) that every cluster must clear before and after customer handover. Establish technology and product direction in collaboration with other tech leads, managers, and senior leadership.
  • Run and instrument real AI workloads as a diagnostic instrument. Stand up and execute distributed training and inference jobs — open-source and customer-representative models — across thousands of accelerators to validate cluster behavior under genuine load rather than synthetic proxies alone, and translate what those runs reveal into fleet-wide fixes.
  • Lead deep diagnosis of large-scale cluster failures and performance regressions , isolating root cause across the full stack: GPU and NIC firmware, PCIe/NVLink topology and NUMA placement, InfiniBand/RoCE fabric health, congestion control and routing, storage and data-loader throughput, scheduler placement, and framework/communication-library behavior. Serve as the final escalation point for the hardest slow-job and stalled-job investigations.
  • Design and build the validation and burn-in systems that qualify nodes, racks, and full pods at scale — NCCL/RCCL collective sweeps, HPL/HPCG and MLPerf-style benchmarks, thermal and power soak tests, straggler and flapping-link detection — and automate them so that qualification is a repeatable pipeline, not a manual campaign.
  • Drive cluster optimization end to end , tuning fabric configuration (adaptive routing, QoS and congestion control, SHARP in-network reduction, rail and topology-aware placement), collective communication libraries and algorithm selection, GPUDirect RDMA and storage paths, and host-level settings (huge pages, IRQ affinity, CPU governors, MIG and driver configuration) to convert raw hardware into delivered throughput.
  • Partner with Infrastructure, Platform, SRE, and customer-facing teams to translate operational and customer performance needs into durable engineering solutions, and to feed diagnostic signal back into provisioning, remediation, and capacity workflows.
  • Build production-grade Python systems and performance

[…]
A, PCIe, IRQ and memory behavior) and strong production Python, plus experience with C/C++ or Go and with configuration management tooling (e.g., Ansible, Terraform).

  • Schedulers: Experience operating and debugging AI workloads under SLURM and/or Kubernetes at scale.

Preferred Qualifications

  • Master's degree or PhD in Engineering, Computer Science, or a related technical field.
  • Experience bringing up and qualifying a greenfield GPU supercluster from first rack to production traffic, including firmware, driver, and topology standardization across a heterogeneous fleet.
  • Direct experience with NVIDIA GPU platforms (H200/GB200/GB300-class), NVLink and NVSwitch domains, DCGM, SHARP, UFM, and the NVIDIA software stack; or equivalent depth on AMD Instinct and ROCm/RCCL.
  • Published or presented benchmark, scaling, or post-mortem work — MLPerf submissions, scaling studies, or public technical write-ups on large-cluster behavior.
  • Experience with advanced observability and monitoring systems (Prometheus, Grafana, OpenTelemetry) applied to high-cardinality GPU and fabric telemetry, including automated anomaly and regression detection.
  • Experience with high-throughput parallel storage (Lustre, GPFS, WEKA, VAST) and with diagnosing data-pipeline-bound training jobs.
  • Familiarity with cloud-native technologies (Kubernetes, Docker), infrastructure-as-code principles, and integration with infrastructure tooling such as DCIMs, NetBox, and bare metal APIs (MAAS, Ironic, IPMI, Redfish).
  • Demonstrated ability to integrate AI tools to optimize/redesign workflows and drive measurable impact (e.g., efficiency gains, quality improvements).
  • Familiarity with SLOs/metrics measurement and logs/telemetry/metrics integration with tools for enhanced operator experience.

For information on how Nscale handles candidate personal data, please see our Employee & Candidate Privacy Notice: Here.

Nscale does not accept unsolicited candidate submissions from recruitment agencies.