Explore doing custom kernels using CuTe DSL like this Nvidia engineer : https://lnkd.in/p/e5dQUGSH
Activity