triton-lang / kernels
☆59Updated last month
Alternatives and similar repositories for kernels:
Users that are interested in kernels are comparing it to the libraries listed below
- extensible collectives library in triton☆73Updated 2 months ago
- Applied AI experiments and examples for PyTorch☆182Updated last week
- ☆173Updated 5 months ago
- Fast low-bit matmul kernels in Triton☆161Updated last week
- Collection of kernels written in Triton language☆75Updated last month
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆91Updated 5 months ago
- An experimental CPU backend for Triton☆65Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆56Updated last month
- ☆94Updated 3 months ago
- ☆56Updated 3 weeks ago
- ☆162Updated this week
- Cataloging released Triton kernels.☆142Updated 3 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆218Updated last month
- ☆79Updated 3 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆28Updated this week
- Fast Hadamard transform in CUDA, with a PyTorch interface☆119Updated 6 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆61Updated 3 months ago
- MLIR-based partitioning system☆49Updated this week
- ☆67Updated 3 years ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆76Updated 3 weeks ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆175Updated this week
- High performance Transformer implementation in C++.☆89Updated 3 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆184Updated last year
- ☆55Updated 6 months ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated 4 months ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆99Updated 2 weeks ago
- Fast Matrix Multiplications for Lookup Table-Quantized LLMs☆199Updated this week
- ☆14Updated last week
- ☆73Updated 2 years ago
- A schedule language for large model training☆141Updated 5 months ago