NVIDIA jobs

NVIDIA seeks a Deep Learning Framework Communications Engineer to integrate advanced communication technologies into AI stacks like PyTorch and TRT-LLM. The role focuses on optimizing multi-GPU communication, scaling across clusters, and collaborating with AI frameworks. Requires 5+ years of software engineering, deep learning framework experience, and proficiency in Python, C++, and CUDA. Located in the United States, remote.

Senior Systems Software Engineer position for NVIDIA's data center platform enablement, focusing on bring-up, integration, validation, and troubleshooting of GPU server platforms. Responsibilities include hardware system design review, simulation/emulation, early prototype debugging, manufacturing support, and root cause analysis. Requires strong computer system and firmware architecture knowledge, CPLD/FPGA experience, SBIOS/BMC/OS interface understanding, and C/C++/bash/python skills. Full-time role at a major AI computing company with emphasis on enterprise server product delivery.

NVIDIA's Earth-2 team is hiring a Senior Scientific Machine Learning Engineer to develop machine learning frameworks (PhysicsNeMo, Earth2Studio) for weather and climate AI. The role involves collaborating on DNN architectures, validating applications, and implementing cutting-edge deep learning techniques including graph networks and diffusion models. Requires 5+ years of experience, BS/MS degree (PhD preferred), strong Python skills, and expertise in deep learning frameworks (PyTorch, TensorFlow, JAX). Familiarity with scientific visualization and HPC programming is a plus. Based in the United States, remote.