Senior Software Engineer, CUTLASS Performance
NVIDIA, US, CA, Santa Clara +4
About the job
NVIDIA's high-performance computing platforms are powering the AI revolution across many applications and industries. Within our software stack, CUTLASS stands out as a popular open-source ecosystem dedicated to high-performance linear algebra and Tensor Core primitives. Since 2017, it has provided the community with C++ and Python abstractions to implement custom matrix multiply (GEMM) and related math and deep learning computations on NVIDIA GPUs.
If you are enthusiastic about performance and eager to help bridge the gap between current performance and what’s theoretically possible, apply to join the CUTLASS team today!
What you'll be doing
- Benchmark the performance of state-of-the-art deep learning models’ inference and training passes to identify key GPU kernel and fusion opportunities.
- Identify gaps between theoretical and realized performance, and suggest software improvements or model adjustments to resolve them.
- Develop tooling to automate the benchmarking, analysis, and performance optimization loop to push the limit of CUTLASS kernel performance within DL networks.