Senior Model Performance Engineer

• 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 our researchers to early engage in ensuring scaling of prototypes of our 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.

• 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.

Come join us at NVIDIA and contribute to a team that is pushing the edges of what can be done in AI and computer vision. We're looking for candidates who are innovative, ambitious, and ready to leave a lasting mark on the world!

We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, sex, gender, gender expression, sexual orientation, age, marital status, veteran status, or disability status. We will ensure that individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us to request accommodation. Widely considered to be one of the technology world's most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology-and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent. As an NVIDIAN, you'll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world. NVIDIA is at the forefront of AI, developing groundbreaking technologies that transform how we interact with the world. Our team is leading the charge in crafting and refining generative models, and 3D reconstruction technologies that enable Physical AI. We are a multidisciplinary group with expertise in machine learning, software engineering, advanced algorithms, and high-performance computing. This is an outstanding opportunity to join a group that is defining the future of AI and computer vision!