SlipstreamJobs tracks this role from the company's public career site. Apply directly on the employer's site.
Lightning AI, the company behind PyTorch Lightning, is seeking a GPU & Compute Infrastructure Engineer to join its Infrastructure Engineering team. The role focuses on owning image management, system diagnostics, and validation across large-scale bare-metal compute infrastructure with emphasis on GPU-enabled systems.
Key responsibilities include:
**Systems, Image & Validation Infrastructure:**
- Own and evolve systems for image management, deployment, and validation across bare-metal infrastructure
- Run and maintain test clusters used for system validation, diagnostics, and cluster bring-up
- Validate firmware, drivers, and OS images across compute and GPU-enabled systems
- Support hardware qualification efforts for next-generation platforms
**GPU Diagnostics & Performance:**
- Own GPU diagnostics and validation workflows across large-scale infrastructure
- Diagnose and resolve complex issues across GPUs, drivers, OS, and hardware layers
- Analyze system and GPU performance using tools such as NVIDIA DCGM
- Identify failure patterns and drive improvements in system stability and validation coverage
You will work at the intersection of hardware, systems, and software—developing automation, improving reliability, and enabling efficient cluster bring-up for AI/ML and HPC workloads. This role is critical to ensuring that Lightning AI's infrastructure is consistent, performant, and ready to support demanding AI workloads.
Lightning AI combines developer-first software with cost-efficient, large-scale compute through its merger with Voltage Park. The company serves solo researchers, startups, and large enterprises globally, with offices in NYC, San Francisco, Seattle, and London. It is backed by Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.
The role is based in one of the company's hubs (NYC, SF, Seattle, or London) with a minimum of 2 in-office days per week and occasional team and company offsites.
