Senior Datacenter Model Engineer

Responsibilities

• Build performance modeling and prediction tools for AI workloads at Data-center scale
• Develop production tools and workflows used by multiple teams both within NVIDIA and its customers
• Automate workflows including search for the most efficient configurations over millions of parameters
• Partner with HW and SW architects to propose new features or improve existing features with real world use cases

• BS+ in Computer Science or related (or equivalent experience) and 5+ years of software development
• Strong software skills in design, coding (C++ and Python), analytical, and debugging
• Good understanding of Deep Learning frameworks like PyTorch and Tensor Flow, distributed training and inference
• Knowledge of GPU cluster job scheduling (Slurm or Kubernetes), storage and networking
• Experience with NVIDIA GPUs, CUDA Programming, and Networking
• Motivated self-starter with strong problem-solving skills and customer-facing communication skills
• Passion for continuous learning. Ability to work concurrently with multiple global groups

• Proven SW engineering experience experience in deploying SW at Dataceter scale
• Solid experience in large AI job performance analysis for training/inference workload
• Knowledge of Linux device drivers and/or compiler implementation
• Knowledge of GPU and/or CPU architecture and general computer architecture principles

Base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is $152,000 - $241,500 USD for Level 3, and $184,000 - $287,500 USD for Level 4.

You will also be eligible for equity and benefits.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive 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.