We are an AI research group led by Prof. Tri Dao
11
公共仓库
27,475
总星标
1,489
关注者
Dao-AILab 是一个专注于人工智能研究的组织,其公共 GitHub 存在展示了多个使用 Python、Cuda、C、C++ 和 HTML 的项目。该组织的知名项目包括 flash-attention 和 quack,涵盖了从高效注意力机制到多种有趣内核的广泛领域。
Fast and memory-efficient exact attention
A Quirky Assortment of CuTe Kernels
Causal depthwise conv1d in CUDA, with a PyTorch interface
Accelerating MoE with IO and Tile-aware Optimizations
Fast Hadamard transform in CUDA, with a PyTorch interface
Fast Polar Decomposition for Muon
此仓库未提供描述。
此仓库未提供描述。
此仓库未提供描述。
CUDA Templates for Linear Algebra Subroutines
此仓库未提供描述。
Dao-AILab 在 GitHub 上构建了多个与人工智能相关的项目,包括 flash-attention 和 quack。这些项目涉及高效的注意力机制和各种有趣的内核,展示了组织在 AI 领域的研究能力。
Dao-AILab 主要使用 Python、Cuda、C、C++ 和 HTML 进行开发。这些语言支持其在深度学习和高性能计算方面的研究和项目开发。
是的,Dao-AILab 的代码库是公开的。这使得其他研究人员和开发者能够访问其项目,促进了开放科学和协作研究。