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 19k 1.9k

  2. causal-conv1d causal-conv1d Public

    Causal depthwise conv1d in CUDA, with a PyTorch interface

    Cuda 548 125

  3. quack quack Public

    A Quirky Assortment of CuTe Kernels

    Python 408 32

  4. fast-hadamard-transform fast-hadamard-transform Public

    Fast Hadamard transform in CUDA, with a PyTorch interface

    C 219 31

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

    Python 123 2

  6. gemm-cublas gemm-cublas Public

    Python 22

Repositories

Showing 7 of 7 repositories
  • flash-attention Public

    Fast and memory-efficient exact attention

    Dao-AILab/flash-attention’s past year of commit activity
    Python 18,997 BSD-3-Clause 1,905 810 78 Updated Aug 20, 2025
  • causal-conv1d Public

    Causal depthwise conv1d in CUDA, with a PyTorch interface

    Dao-AILab/causal-conv1d’s past year of commit activity
    Cuda 548 BSD-3-Clause 125 28 9 Updated Aug 20, 2025
  • quack Public

    A Quirky Assortment of CuTe Kernels

    Dao-AILab/quack’s past year of commit activity
    Python 408 Apache-2.0 32 6 1 Updated Aug 18, 2025
  • cutlass Public Forked from NVIDIA/cutlass

    CUDA Templates for Linear Algebra Subroutines

    Dao-AILab/cutlass’s past year of commit activity
    C++ 1 1,399 0 0 Updated Jun 8, 2025
  • grouped-latent-attention Public
    Dao-AILab/grouped-latent-attention’s past year of commit activity
    Python 123 MIT 2 4 0 Updated May 30, 2025
  • gemm-cublas Public
    Dao-AILab/gemm-cublas’s past year of commit activity
    Python 22 Apache-2.0 0 0 0 Updated May 5, 2025
  • fast-hadamard-transform Public

    Fast Hadamard transform in CUDA, with a PyTorch interface

    Dao-AILab/fast-hadamard-transform’s past year of commit activity
    C 219 BSD-3-Clause 31 7 3 Updated May 24, 2024

Most used topics

Loading…