AI and ML HPC Cluster Engineer

AI and ML HPC Cluster Engineer DIA is a pioneer in accelerated computing, known for inventing the GPU and driving breakthroughs in gaming, computer graphics, high‑performance computing, and artificial intelligence. Our technology powers everything from generative AI to autonomous systems, and we continue to shape the future of computing through innovation and collaboration. Within this mission, our Managed AI Superclusters (MARS) team builds and scales the infrastructure, platforms, and tools that enable researchers and engineers to develop the next generation of AI/ML systems.

By joining us, you’ll help design solutions that power some of the world’s most advanced computing workloads.

• Support day‑to‑day operations of production on‑premises and multi‑cloud AI/HPC clusters, ensuring system health, user satisfaction, and efficient resource utilization.
• Directly administer internal research clusters, conduct upgrades, incident response, and reliability improvements.
• Develop and improve our ecosystem around GPU‑accelerated computing, including developing scalable automation solutions.
• Maintain heterogeneous AI/ML clusters on‑premises and in the cloud.
• Support our researchers to run their workloads, including performance analysis and optimizations.
• Analyze and optimize cluster efficiency, job fragmentation, and GPU waste to meet internal SLA targets.
• Support root cause analysis and suggest corrective action. Proactively find and fix issues before they occur.
• Triage and support postmortems for reliability incidents affecting users or infrastructure.
• Participate in a shared on‑call rotation supported by strong automation, clear paths for responding to critical issues, and well‑defined incident workflows.

• Bachelor’s degree in Computer Science, Electrical Engineering or related field or equivalent experience.
• Minimum 2 years of experience administering multi‑node compute infrastructure.
• Background in managing AI/HPC job schedulers like Slurm, K8s, PBS, RTDA, BCM (formerly known as Bright), or LSF.
• Proficient in administering CentOS/RHEL and/or Ubuntu Linux distributions.
• Proven understanding of cluster configuration management tools (Ansible, Puppet, Salt, etc.), container technologies (Docker, Singularity, Podman, Shifter, Charlie cloud), Python programming, and bash scripting.
• Passion for continual learning and staying ahead of emerging technologies and effective approaches in the HPC and AI/ML infrastructure fields.

• Background with NVIDIA GPUs, CUDA Programming, NCCL, and MLPerf benchmarking.
• Experience with AI/ML concepts, algorithms, models, and frameworks (PyTorch, Tensorflow).
• Experience with Infini Band with IBOP and RDMA.
• Understanding of fast, distributed storage systems such as Lustre and GPFS for AI/HPC workloads.
• Applied knowledge in AI/HPC workflows that involve MPI.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 120,000 USD - 189,750 USD. You will also be eligible for equity and benefits.

Entry level

Full‑time

Computer Hardware Manufacturing, Software Development, and Computers and Electronics Manufacturing

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.