pytorch-labs / superblock
A block oriented training approach for inference time optimization.
☆32Updated 5 months ago
Alternatives and similar repositories for superblock:
Users that are interested in superblock are comparing it to the libraries listed below
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆75Updated this week
- ☆157Updated last year
- Fast low-bit matmul kernels in Triton☆187Updated last week
- A library for unit scaling in PyTorch☆118Updated last month
- This repository contains the experimental PyTorch native float8 training UX☆219Updated 5 months ago
- ☆96Updated 4 months ago
- extensible collectives library in triton☆76Updated 3 months ago
- Experiment of using Tangent to autodiff triton☆74Updated 11 months ago
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 10 months ago
- ☆30Updated 5 months ago
- Applied AI experiments and examples for PyTorch☆211Updated this week
- Cataloging released Triton kernels.☆156Updated last week
- Fast Hadamard transform in CUDA, with a PyTorch interface☆133Updated 7 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 6 months ago
- ☆170Updated last week
- ☆52Updated last week
- ☆57Updated 7 months ago
- ☆83Updated 7 months ago
- ☆64Updated 2 months ago
- Patch convolution to avoid large GPU memory usage of Conv2D☆81Updated 7 months ago
- Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops☆24Updated 10 months ago
- Fast Matrix Multiplications for Lookup Table-Quantized LLMs☆219Updated this week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆93Updated 6 months ago
- ☆56Updated 3 months ago
- Collection of kernels written in Triton language☆90Updated 2 months ago
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆114Updated 10 months ago
- ☆34Updated last month
- Triton-based implementation of Sparse Mixture of Experts.☆192Updated last month
- The simplest but fast implementation of matrix multiplication in CUDA.☆34Updated 5 months ago
- ☆21Updated 2 months ago