Alternatives and similar repositories for sten:

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

• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated 11 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆83Updated 5 months ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆150Updated 9 months ago
• topal-team / rockmate
  ☆35Updated 2 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆76Updated last year
• exists-forall / striped_attention
  ☆38Updated last year
• gpauloski / kfac-pytorch
  Distributed K-FAC Preconditioner for PyTorch
  ☆85Updated this week
• Jokeren / triton-samples
  ☆26Updated last month
• triton-lang / kernels
  ☆71Updated 3 months ago
• stanford-futuredata / stk
  ☆100Updated 6 months ago
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆65Updated this week
• olcf / NVIDIA-tensor-core-examples
  ☆17Updated 5 years ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆99Updated this week
• microsoft / TileFusion
  TileFusion is a highly efficient kernel template library designed to elevate the level of abstraction in CUDA C for processing tiles.
  ☆61Updated this week
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆38Updated 2 years ago
• manishucsd / py-codegen
  ☆15Updated 5 months ago
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆23Updated 2 months ago
• ScalingIntelligence / CATS
  ☆23Updated 3 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆108Updated 2 weeks ago
• awslabs / ratex
  ☆23Updated 3 months ago
• spcl / substation
  Research and development for optimizing transformers
  ☆125Updated 4 years ago
• feifeibear / DPSKV3MFU
  Estimate MFU for DeepSeekV3
  ☆18Updated 2 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆65Updated 4 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.
  ☆43Updated 7 months ago
• emalach / LinearLM
  Code for the paper: https://arxiv.org/pdf/2309.06979.pdf
  ☆19Updated 7 months ago
• zhisbug / Cavs
  Cavs: An Efficient Runtime System for Dynamic Neural Networks
  ☆14Updated 4 years ago
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆75Updated 2 months ago
• brightlaboratory / polydl
  ☆12Updated 3 years ago
• Aleph-Alpha / NeurIPS-WANT-submission-efficient-parallelization-layouts
  ☆22Updated last year
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 7 months ago