Dao-AILab
Githubhttps://tridao.me/

代码库

Dao-AILab/causal-conv1d
934
Causal depthwise conv1d in CUDA, with a PyTorch interface
Cuda
Dao-AILab/flash-attention
24.7k
Fast and memory-efficient exact attention
Python