spcl / stenLinks
Sparsity support for PyTorch
☆37Updated 7 months ago
Alternatives and similar repositories for sten
Users that are interested in sten are comparing it to the libraries listed below
Sorting:
- extensible collectives library in triton☆90Updated 7 months ago
 - ☆41Updated last year
 - ☆28Updated 9 months ago
 - A library of GPU kernels for sparse matrix operations.☆275Updated 4 years ago
 - ☆20Updated 5 years ago
 - Distributed SDDMM Kernel☆11Updated 3 years ago
 - TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆100Updated 4 months ago
 - ☆112Updated last year
 - This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆71Updated last month
 - ☆23Updated 2 months ago
 - ☆93Updated 11 months ago
 - A bunch of kernels that might make stuff slower 😉☆63Updated this week
 - A Data-Centric Compiler for Machine Learning☆85Updated last year
 - ☆16Updated last year
 - SparseTIR: Sparse Tensor Compiler for Deep Learning☆138Updated 2 years ago
 - ☆58Updated this week
 - Test suite for probing the numerical behavior of NVIDIA tensor cores☆41Updated last year
 - Parallel framework for training and fine-tuning deep neural networks☆65Updated last week
 - ☆35Updated last week
 - Automatic differentiation for Triton Kernels☆13Updated 2 months ago
 - ☆36Updated 10 months ago
 - A schedule language for large model training☆151Updated 2 months ago
 - How to ensure correctness and ship LLM generated kernels in PyTorch☆111Updated this week
 - Research and development for optimizing transformers☆131Updated 4 years ago
 - A Python library transfers PyTorch tensors between CPU and NVMe☆120Updated 11 months ago
 - Training neural networks in TensorFlow 2.0 with 5x less memory☆136Updated 3 years ago
 - CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆50Updated 7 years ago
 - Experiment of using Tangent to autodiff triton☆80Updated last year
 - ☆11Updated 4 years ago
 - Personal solutions to the Triton Puzzles☆20Updated last year