Skip to content

@Dao-AILab Dao AI Lab

Change the repository type filter

All

    Repositories list

    4 repositories

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.4k•15k•639•49•Updated Feb 10, 2025Feb 10, 2025

    • cutlass

      Public
      CUDA Templates for Linear Algebra Subroutines
      C++
      •
      Other
      •1.1k•1•0•0•Updated Feb 8, 2025Feb 8, 2025

    • causal-conv1d

      Public
      Causal depthwise conv1d in CUDA, with a PyTorch interface
      Cuda
      •
      BSD 3-Clause "New" or "Revised" License
      •74•385•19•5•Updated Dec 6, 2024Dec 6, 2024

    • fast-hadamard-transform

      Public
      Fast Hadamard transform in CUDA, with a PyTorch interface
      C
      •
      BSD 3-Clause "New" or "Revised" License
      •20•138•4•2•Updated May 24, 2024May 24, 2024