Requirements:

MSc or PhD in Computer Science, Computer Engineering, or a related field with 8+ years of experience in machine learning, computer vision, or equivalent experience.
A passion in optimizing AI models and digging into low-level GPU processes to understand how every layer works.
Strong programming skills in Python, C++ and CUDA (Slang is a plus).
Experience with running and optimizing distributed training on a multi-node multi-GPU system.
Strong software engineering fundamentals, including source control, testing/validation, and containerization.
Excellent communication and interpersonal skills, with the ability to work effectively in a dynamic, product- and research-focused team.

Ways to Stand Out from the Crowd:

Strong coding architecture skills proven through contributions to large internal or open-source projects.
Hands-on experience with NCCL, MPI, or UCX communication libraries.
Experience with advanced CUDA and Slang optimization for graphics or vision applications.
A history of multidisciplinary creativity and innovation, such as experience with hardware and software projects in graphics or robotics.
Experience in robotic systems such as autonomous vehicles or humanoid robotics.

Profile model training and inference to identify bottlenecks regarding efficiency, latency, and memory usage.
Optimize models with respect to GPU utilization and throughput by improving kernel efficiency.
Partner with researchers to early engage in ensuring scaling of prototypes of the latest models for generative video creation, segmentation, video understanding, tracking, 3D reconstruction, and more.
Collaborate within a large and dynamic codebase building robust and efficient models for synthetic data and world generation.
Interact closely with different research, performance, and product teams at NVIDIA.
Contribute to NVIDIA NuRec and other core NVIDIA products and libraries.

Highly competitive salaries.
Comprehensive benefits package.
Equal opportunity employer promoting diversity in the workplace.
Accommodation provided for individuals with disabilities during the application process and employment.