Description
We are seeking a Senior Software Engineer to join our CUTLASS team. As a member of this team, you will 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. You will also contribute to the advancement of the MLIR-based backend compiler stack for the CUTLASS Python DSL by designing dialects and associated compiler passes. Additionally, you will author example kernels utilizing CUTLASS abstractions to showcase the use of novel GPU hardware features that are crucial for achieving high performance.
The ideal candidate will have a strong proficiency in C++ programming and software design, including debugging, performance evaluation, and testing. They will also have 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 is also required.
We offer a competitive salary and benefits package, as well as opportunities for professional growth and development. If you are passionate about designing abstractions for Tensor Core and related GPU hardware features in MLIR, Python, and C++, apply to join our team today!