Popular repositories Loading

1. flash-attention flash-attention Public

   Fast and memory-efficient exact attention

   Python 17.9k 1.7k

2. causal-conv1d causal-conv1d Public

   Causal depthwise conv1d in CUDA, with a PyTorch interface

   Cuda 492 105

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

   Fast Hadamard transform in CUDA, with a PyTorch interface

   C 198 27

4. grouped-latent-attention grouped-latent-attention Public

   Python 113 2

5. gemm-cublas gemm-cublas Public

   Python 21

6. cutlass cutlass Public

   Forked from NVIDIA/cutlass

   CUDA Templates for Linear Algebra Subroutines

   C++ 1