Official implementation of FlashAttention, a memory-efficient exact attention kernel for training and inference of Transformer models on GPUs.
25.2K3.2KPythonBSD-3-ClauseUpdated 9 hours ago
#attention
Curated GitHub repositories related to attention, with AI summaries and key stats.
Official implementation of FlashAttention, a memory-efficient exact attention kernel for training and inference of Transformer models on GPUs.
SGLang is an open-source inference framework for serving large language, vision-language, and diffusion models on many GPU, TPU, and CPU platforms.