Senior ML Systems Engineer - GPU PyTorch & GNNs (Equity)

NVIDIA is seeking a highly experienced Senior Software Engineer to join a team building large-scale machine learning solutions, including Graph Neural Networks, Tabular Foundation Models, and ensemble models, with a focus on efficient training and inference on NVIDIA GPU infrastructure.

You will partner with researchers and engineers to accelerate PyTorch-based frameworks, optimize CUDA-X integrations, and mentor teammates while delivering robust, scalable software for customers leveraging