Alternatives and similar repositories for sten:

Users that are interested in sten are comparing it to the libraries listed below

• cchan / tccl
  extensible collectives library in triton
  ☆84Updated 2 weeks ago
• exists-forall / striped_attention
  ☆38Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• gpauloski / kfac-pytorch
  Distributed K-FAC Preconditioner for PyTorch
  ☆85Updated this week
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated last year
• topal-team / rockmate
  ☆36Updated 4 months ago
• triton-lang / kernels
  ☆76Updated 5 months ago
• stanford-futuredata / stk
  ☆103Updated 7 months ago
• Jokeren / triton-samples
  ☆27Updated 3 months ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆168Updated 10 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆110Updated this week
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆67Updated last month
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆211Updated 4 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆81Updated 5 months ago
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆38Updated 2 years ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆118Updated last week
• thu-ml / Jetfire-INT8Training
  ☆40Updated 8 months ago
• ScalingIntelligence / CATS
  ☆24Updated 5 months ago
• BradMcDanel / sdgp
  ☆10Updated 3 years ago
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆51Updated 7 years ago
• zhisbug / Cavs
  Cavs: An Efficient Runtime System for Dynamic Neural Networks
  ☆14Updated 4 years ago
• insuhan / hyper-attn
  ☆81Updated last year
• hgyhungry / ShflBW_Sparse_NN
  ☆14Updated 2 years ago
• Infini-AI-Lab / Sirius
  Sirius, an efficient correction mechanism, which significantly boosts Contextual Sparsity models on reasoning tasks while maintaining its…
  ☆21Updated 7 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆80Updated last week
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆45Updated 9 months ago
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆58Updated last month
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆72Updated 7 months ago
• LiuXiaoxuanPKU / GACT-ICML
  ☆42Updated 2 years ago
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆134Updated 2 years ago