8000 Dao AI Lab · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
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 17.6k 1.7k

  2. causal-conv1d causal-conv1d Public

    Causal depthwise conv1d in CUDA, with a PyTorch interface

    Cuda 468 103

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

    Fast Hadamard transform in CUDA, with a PyTorch interface

    C 192 24

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

    Python 59

  5. gemm-cublas gemm-cublas Public

    Python 20

  6. cutlass cutlass Public

    Forked from NVIDIA/cutlass

    CUDA Templates for Linear Algebra Subroutines

    C++ 1

Repositories

Showing 6 of 6 repositories

Top languages

Loading…

Most used topics

Loading…
0