Alternatives and similar repositories for sten:

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

• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• topal-team / rockmate
  ☆36Updated 4 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆85Updated last month
• exists-forall / striped_attention
  ☆38Updated last year
• gpauloski / kfac-pytorch
  Distributed K-FAC preconditioner for PyTorch
  ☆85Updated this week
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆64Updated last year
• stanford-futuredata / stk
  ☆104Updated 8 months ago
• insuhan / hyper-attn
  ☆81Updated last year
• Jokeren / triton-samples
  ☆27Updated 3 months ago
• mayank31398 / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆40Updated this week
• DerrickYLJ / TidalDecode
  [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention
  ☆35Updated 2 weeks ago
• manishucsd / py-codegen
  ☆16Updated 7 months ago
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆38Updated 9 months ago
• olcf / NVIDIA-tensor-core-examples
  ☆18Updated 5 years ago
• Aleph-Alpha / NeurIPS-WANT-submission-efficient-parallelization-layouts
  ☆22Updated last year
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆122Updated this week
• LiuXiaoxuanPKU / GACT-ICML
  ☆42Updated 2 years ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆183Updated 11 months ago
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆38Updated 2 years ago
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆58Updated last month
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆68Updated 2 months ago
• 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
• triton-lang / kernels
  ☆78Updated 6 months ago
• HazyResearch / fly
  ☆204Updated 2 years ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆88Updated last week
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆73Updated 8 months ago
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆102Updated this week
• hdong920 / GRIFFIN
  ☆36Updated 8 months ago
• DS3Lab / CocktailSGD
  ☆27Updated last year
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 7 months ago