Skip to content
@Dao-AILab

Dao AI Lab

We are an AI research group led by Prof. Tri Dao

Popular repositories Loading

  1. flash-attention flash-attention Public

    Fast and memory-efficient exact attention

    Python 18.2k 1.8k

  2. causal-conv1d causal-conv1d Public

    Causal depthwise conv1d in CUDA, with a PyTorch interface

    Cuda 503 110

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

    Fast Hadamard transform in CUDA, with a PyTorch interface

    C 201 27

  4. quack quack Public

    A Quirky Assortment of CuTe Kernels

    Python 124 4

  5. grouped-latent-attention grouped-latent-attention Public

    Python 116 2

  6. gemm-cublas gemm-cublas Public

    Python 21

Repositories

Showing 7 of 7 repositories

Most used topics

Loading…