Job Summary:
NVIDIA is a leader in high-performance computing platforms driving the AI revolution across various industries. They are seeking a Senior Software Engineer to develop core components of the CUTLASS platform, contributing to high-performance linear algebra and Tensor Core primitives and collaborating with other teams to enhance GPU programming models.
Responsibilities:
โข Develop core components of the CUTLASS platform including Tensor Core MMAs, copies, synchronization barriers, schedulers, and other GPU hardware features in CUDA C++ and CUTLASS Python DSL.
โข Contribute to the advancement of the MLIR-based backend compiler stack for the CUTLASS Python DSL by designing dialects and associated compiler passes.
โข Author example kernels utilizing CUTLASS abstractions to showcase the use of novel GPU hardware features that are crucial for achieving high performance.
โข Collaborate with GPU architecture, CUDA, and NVVM/PTX compiler teams to provide feedback on programming models and to assess the performance of future GPU hardware features.
Qualifications:
Required:
โข 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 working with high-performance code generation and knowledge of compiler transformations and optimizations.
โข A deep understanding of computer architecture and parallel computing programming models.
Preferred:
โข Experience writing high-performance kernels at low levels of abstractions like NVVM/ PTX for GPUs or other similar parallel processing architectures.
โข Hands-on compiler design experience, particularly in MLIR.
โข Understanding of deep learning models, algorithms, and frameworks.
Company:
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. Founded in 1993, the company is headquartered in Santa Clara, USA, with a team of 10001+ employees. The company is currently Late Stage.