Skip to content

@Dao-AILab Dao AI Lab

Change the repository type filter

All

    Repositories list

    8 repositories

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •2.2k•21k•916•92•Updated Dec 18, 2025Dec 18, 2025

    • sonic-moe

      Public
      Accelerating MoE with IO and Tile-aware Optimizations
      Python
      •
      Apache License 2.0
      •15•343•1•0•Updated Dec 18, 2025Dec 18, 2025

    • quack

      Public
      A Quirky Assortment of CuTe Kernels
      Python
      •
      Apache License 2.0
      •64•700•14•1•Updated Dec 16, 2025Dec 16, 2025

    • causal-conv1d

      Public
      Causal depthwise conv1d in CUDA, with a PyTorch interface
      Cuda
      •
      BSD 3-Clause "New" or "Revised" License
      •147•677•33•11•Updated Oct 20, 2025Oct 20, 2025

    • fast-hadamard-transform

      Public
      Fast Hadamard transform in CUDA, with a PyTorch interface
      C
      •
      BSD 3-Clause "New" or "Revised" License
      •49•267•8•2•Updated Oct 19, 2025Oct 19, 2025

    • cutlass

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

    • grouped-latent-attention

      Public
      Python
      •
      MIT License
      •4•133•5•0•Updated May 29, 2025May 29, 2025

    • gemm-cublas

      Public
      Python
      •
      Apache License 2.0
      •1•23•0•0•Updated May 5, 2025May 5, 2025