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 22.2k 2.4k

  2. quack quack Public

    A Quirky Assortment of CuTe Kernels

    Python 798 80

  3. causal-conv1d causal-conv1d Public

    Causal depthwise conv1d in CUDA, with a PyTorch interface

    Cuda 720 152

  4. sonic-moe sonic-moe Public

    Accelerating MoE with IO and Tile-aware Optimizations

    Python 585 52

  5. fast-hadamard-transform fast-hadamard-transform Public

    Fast Hadamard transform in CUDA, with a PyTorch interface

    C 284 52

  6. grouped-latent-attention grouped-latent-attention Public

    Python 131 4

Repositories

Showing 8 of 8 repositories
  • quack Public

    A Quirky Assortment of CuTe Kernels

    Dao-AILab/quack’s past year of commit activity
    Python 798 Apache-2.0 80 18 2 Updated Feb 13, 2026
  • flash-attention Public

    Fast and memory-efficient exact attention

    Dao-AILab/flash-attention’s past year of commit activity
    Python 22,242 BSD-3-Clause 2,382 943 115 Updated Feb 14, 2026
  • sonic-moe Public

    Accelerating MoE with IO and Tile-aware Optimizations

    Dao-AILab/sonic-moe’s past year of commit activity
    Python 585 Apache-2.0 52 13 4 Updated Feb 13, 2026
  • causal-conv1d Public

    Causal depthwise conv1d in CUDA, with a PyTorch interface

    Dao-AILab/causal-conv1d’s past year of commit activity
    Cuda 720 BSD-3-Clause 152 35 11 Updated Jan 12, 2026
  • 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 284 BSD-3-Clause 52 9 2 Updated Oct 19, 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,692 0 0 Updated Jun 8, 2025
  • grouped-latent-attention Public
    Dao-AILab/grouped-latent-attention’s past year of commit activity
    Python 131 MIT 4 5 0 Updated May 29, 2025
  • gemm-cublas Public
    Dao-AILab/gemm-cublas’s past year of commit activity
    Python 22 Apache-2.0 1 0 0 Updated May 5, 2025

Most used topics

Loading…