Skip to content

@Dao-AILab Dao AI Lab

All

    Repositories list

    12 repositories

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •3.1k•25k•1k•282•Updated Sep 13, 2026Sep 13, 2026

    • quack

      Public
      A Quirky Assortment of CuTe Kernels
      Python
      •
      Apache License 2.0
      •155•1.1k•14•21•Updated Sep 13, 2026Sep 13, 2026

    • dao-ailab.github.io

      Public
      HTML
      •
      MIT License
      •2•2•0•0•Updated Sep 9, 2026Sep 9, 2026

    • sonic-moe

      Public
      Accelerating MoE with IO and Tile-aware Optimizations
      Python
      •
      Apache License 2.0
      •105•765•9•7•Updated Aug 29, 2026Aug 29, 2026

    • causal-conv1d

      Public
      Causal depthwise conv1d in CUDA, with a PyTorch interface
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •210•952•43•21•Updated Aug 20, 2026Aug 20, 2026

    • gambit-parallel-reasoning

      Public
      Thought-level beam search for parallel reasoning via hardware-aware pruning and adaptive compute reallocation
      Python
      •
      Apache License 2.0
      •0•7•1•0•Updated Aug 20, 2026Aug 20, 2026

    • gram-newton-schulz

      Public
      Fast Polar Decomposition for Muon
      Python
      •15•182•5•1•Updated Jul 2, 2026Jul 2, 2026

    • AI-workflow

      Public
      3•73•1•0•Updated Mar 24, 2026Mar 24, 2026

    • fast-hadamard-transform

      Public
      Fast Hadamard transform in CUDA, with a PyTorch interface
      C
      •
      BSD 3-Clause "New" or "Revised" License
      •76•349•8•4•Updated Mar 10, 2026Mar 10, 2026

    • cutlass

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

    • grouped-latent-attention

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

    • gemm-cublas

      Public
      Python
      •
      Apache License 2.0
      •1•22•0•0•Updated May 5, 2025May 5, 2025
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.