Popular repositories Loading

1. flash-attention flash-attention Public

   Fast and memory-efficient exact attention

   Python 17.5k 1.7k

2. causal-conv1d causal-conv1d Public

   Causal depthwise conv1d in CUDA, with a PyTorch interface

   Cuda 462 98

3. fast-hadamard-transform fast-hadamard-transform Public

   Fast Hadamard transform in CUDA, with a PyTorch interface

   C 188 24

4. gemm-cublas gemm-cublas Public

   Python 19

5. cutlass cutlass Public

   Forked from NVIDIA/cutlass

   CUDA Templates for Linear Algebra Subroutines

   C++ 1