HazyResearch / aisys-building-blocksView external linksLinks
Building blocks for foundation models.
☆601Jan 3, 2024Updated 2 years ago
Alternatives and similar repositories for aisys-building-blocks
Users that are interested in aisys-building-blocks are comparing it to the libraries listed below
Sorting:
- Tile primitives for speedy kernels☆3,139Updated this week
- GPU programming related news and material links☆1,967Sep 17, 2025Updated 4 months ago
- Understand and test language model architectures on synthetic tasks.☆252Jan 12, 2026Updated last month
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- An ML Systems Onboarding list☆984Jan 24, 2025Updated last year
- Experiment of using Tangent to autodiff triton☆82Jan 22, 2024Updated 2 years ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆326Updated this week
- Cataloging released Triton kernels.☆294Sep 9, 2025Updated 5 months ago
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆342Dec 28, 2024Updated last year
- ☆51Jan 28, 2024Updated 2 years ago
- Applied AI experiments and examples for PyTorch☆317Aug 22, 2025Updated 5 months ago
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,379Updated this week
- Fast low-bit matmul kernels in Triton☆429Feb 1, 2026Updated 2 weeks ago
- Framework to reduce autotune overhead to zero for well known deployments.☆96Sep 19, 2025Updated 4 months ago
- A Quirky Assortment of CuTe Kernels☆798Updated this week
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆106Jun 28, 2025Updated 7 months ago
- Material for gpu-mode lectures☆5,726Feb 1, 2026Updated 2 weeks ago
- Distributed Compiler based on Triton for Parallel Systems☆1,350Feb 9, 2026Updated last week
- Helpful tools and examples for working with flex-attention☆1,127Feb 8, 2026Updated last week
- Puzzles for learning Triton☆2,296Nov 18, 2024Updated last year
- Benchmark tests supporting the TiledCUDA library.☆18Nov 19, 2024Updated last year
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆443Feb 4, 2026Updated last week
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark + Toolkit with Torch -> CUDA (+ more DSLs)☆806Jan 20, 2026Updated 3 weeks ago
- 🎬 3.7× faster video generation E2E 🖼️ 1.6× faster image generation E2E ⚡ ColumnSparseAttn 9.3× vs FlashAttn‑3 💨 ColumnSparseGEMM 2.5× …☆102Sep 8, 2025Updated 5 months ago
- Efficient Triton Kernels for LLM Training☆6,141Updated this week
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆248Jun 6, 2025Updated 8 months ago
- FlashInfer: Kernel Library for LLM Serving☆4,935Feb 10, 2026Updated last week
- DeeperGEMM: crazy optimized version☆74May 5, 2025Updated 9 months ago
- Accelerated First Order Parallel Associative Scan☆196Jan 7, 2026Updated last month
- Flash Attention in ~100 lines of CUDA (forward pass only)☆1,068Dec 30, 2024Updated last year
- A bibliography and survey of the papers surrounding o1☆1,212Nov 16, 2024Updated last year
- What would you do with 1000 H100s...☆1,153Jan 10, 2024Updated 2 years ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆596Aug 12, 2025Updated 6 months ago
- A PyTorch native platform for training generative AI models☆5,069Updated this week
- ☆65Apr 26, 2025Updated 9 months ago
- ☆288Updated this week
- Development repository for the Triton language and compiler☆18,429Updated this week
- ☆261Jul 11, 2024Updated last year
- Perplexity GPU Kernels☆562Nov 7, 2025Updated 3 months ago