AI工具

cutlass

NVIDIA/cutlass

CUTLASS是NVIDIA开源的CUDA C++模板抽象库,用于实现高性能矩阵乘法(GEMM)及相关线性代数运算,支持多种精度和架构,并提供Python DSL简化CUDA内核开发。