Teams at NVIDIA

As a Principal Software Engineer, you will lead AI networking system transformations, manage complex customer engagements, and influence NVIDIA's networking technologies.

NVIDIA seeks a Senior Software Engineer to enhance its AI platform using JAX, focusing on performance optimization in deep learning frameworks while collaborating with applied researchers. The role requires strong programming, system design skills, and a proven ability to develop customer-facing solutions.

Design and implement compiler IR, programming models, and optimizations for GPU-based AI inference; define public APIs; collaborate with hardware and framework teams to optimize performance, kernel generation, and build-time for Ahead-of-Time and Just-in-Time deployments.