Pinned Loading

1. TransformerEngine TransformerEngine Public

   Forked from NVIDIA/TransformerEngine

   A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilizatio…

   Python

2. surgical_ssms surgical_ssms Public

   State Space Models for Surgical Phase Recognition

   Python

3. cutlass cutlass Public

   Forked from NVIDIA/cutlass

   CUDA Templates for Linear Algebra Subroutines

   C++

4. flash-attention flash-attention Public

   Forked from Dao-AILab/flash-attention

   Fast and memory-efficient exact attention

   Python