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
  ☆88Updated 4 months ago
• Jokeren / triton-samples
  ☆28Updated 6 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆93Updated last month
• exists-forall / striped_attention
  ☆39Updated last year
• IST-DASLab / FP-Quant
  ☆21Updated last week
• PASSIONLab / distributed_sddmm
  Distributed SDDMM Kernel
  ☆11Updated 3 years ago
• stanford-futuredata / stk
  ☆108Updated 11 months ago
• open-lm-engine / flash-model-architectures
  A bunch of kernels that might make stuff slower 😉
  ☆56Updated last week
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆270Updated 4 years ago
• triton-lang / kernels
  ☆85Updated 9 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆79Updated 2 weeks ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆142Updated 4 months ago
• awslabs / ratex
  ☆23Updated 8 months ago
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆212Updated this week
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆137Updated 2 years ago
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆63Updated 4 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆40Updated 2 years ago
• daniel-geon-park / triton_bwd
  Automatic differentiation for Triton Kernels
  ☆11Updated this week
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆118Updated 8 months ago
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆40Updated last year
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆132Updated 3 years ago
• manishucsd / py-codegen
  ☆16Updated 10 months ago
• topal-team / rockmate
  ☆36Updated 8 months ago
• meta-pytorch / kraken
  Triton-based Symmetric Memory operators and examples
  ☆23Updated last week
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆52Updated last month
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆51Updated 7 years ago
• ademeure / cuda-side-boost
  ☆41Updated 3 months ago
• spcl / substation
  Research and development for optimizing transformers
  ☆129Updated 4 years ago
• YashasSamaga / ConvolutionBuildingBlocks
  GEMM and Winograd based convolutions using CUTLASS
  ☆26Updated 5 years ago