Thanks to visit codestin.com
Credit goes to github.com

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 21.7k 2.3k

  2. quack quack Public

    A Quirky Assortment of CuTe Kernels

    Python 751 73

  3. causal-conv1d causal-conv1d Public

    Causal depthwise conv1d in CUDA, with a PyTorch interface

    Cuda 705 150

  4. sonic-moe sonic-moe Public

    Accelerating MoE with IO and Tile-aware Optimizations

    Python 547 44

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

    Fast Hadamard transform in CUDA, with a PyTorch interface

    C 274 50

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

    Python 133 4

Repositories

Showing 8 of 8 repositories

Top languages

Loading…

Most used topics

Loading…