NVIDIA is seeking a passionate developer to join the CUTLASS team, focusing on high-performance computing platforms that drive the AI revolution. The role involves developing and optimizing Tensor Core-based deep learning kernels using CUTLASS in CUDA C++ and Python. Candidates will collaborate with various teams across NVIDIA to ensure timely delivery of high-performance kernels. This position offers the opportunity to work on cutting-edge technology in a dynamic and inclusive environment.

• Masters or PhD degree in Computer Science, Computer Engineering, or related field or equivalent experience.
• 3+ years of relevant industry experience.
• Strong proficiency in C++ programming and software design, including debugging, performance evaluation, and testing.
• Experience with CUDA, OpenCL, HIP, SYCL, Mojo, Pallas, Triton, Mosaic, Halide, or any general-purpose or domain-specific programming language targeting highly parallel accelerators.
• Deep understanding of computer architecture and some experience working at the assembly level.

• Write Tensor Core-based deep learning kernels such as grouped-GEMM, attention, and convolution using CUTLASS CUDA C++ and Python DSL.
• Optimize kernels for peak throughput on both silicon and software performance simulators.
• Collaborate with teams across NVIDIA including the GPU architecture, NVVM/PTX compiler, CUDA library, and DL frameworks teams to ensure fast, functional, and timely kernel delivery to customers.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.