Skip to content
@Dao-AILab

Dao AI Lab

We are a CS research group led by Prof. Tri Dao

Popular repositories

  1. flash-attention flash-attention Public

    Fast and memory-efficient exact attention

    Python 11.5k 1k

  2. causal-conv1d causal-conv1d Public

    Causal depthwise conv1d in CUDA, with a PyTorch interface

    Cuda 203 39

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

    Fast Hadamard transform in CUDA, with a PyTorch interface

    C 58 5

Repositories

Showing 3 of 3 repositories

People

This organization has no public members. You must be a member to see who’s a part of this organization.

Top languages

Python C Cuda

Most used topics

Loading…