Alternatives and similar repositories for rockmate:

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

• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated 10 months ago
• BradMcDanel / sdgp
  ☆11Updated 2 years ago
• spcl / sten
  Sparsity support for PyTorch
  ☆33Updated last month
• stanford-futuredata / stk
  ☆96Updated 4 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆75Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆74Updated 11 months ago
• ScalingIntelligence / CATS
  ☆23Updated 2 months ago
• cli99 / flops-profiler
  pytorch-profiler
  ☆50Updated last year
• cchan / tccl
  extensible collectives library in triton
  ☆76Updated 3 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 6 months ago
• VITA-Group / Structure-LTH
  [ICML 2022] "Coarsening the Granularity: Towards Structurally Sparse Lottery Tickets" by Tianlong Chen, Xuxi Chen, Xiaolong Ma, Yanzhi Wa…
  ☆31Updated last year
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• pytorch-labs / superblock
  A block oriented training approach for inference time optimization.
  ☆32Updated 5 months ago
• albanD / subclass_zoo
  ☆154Updated 7 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆118Updated last month
• gpauloski / kfac-pytorch
  Distributed K-FAC Preconditioner for PyTorch
  ☆84Updated this week
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆133Updated 7 months ago
• yifuwang / symm-mem-recipes
  ☆27Updated 3 weeks ago
• triton-lang / kernels
  ☆64Updated 2 months ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆130Updated 2 years ago
• yaozhewei / HAP
  ☆43Updated 11 months ago
• AamirRaihan / SWAT
  Official implementation of Neurips 2020 "Sparse Weight Activation Training" paper.
  ☆27Updated 3 years ago
• thu-ml / Jetfire-INT8Training
  ☆30Updated 5 months ago
• andylolu2 / simpleGEMM
  The simplest but fast implementation of matrix multiplication in CUDA.
  ☆34Updated 5 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆90Updated 2 months ago
• mobiusml / gemlite
  Fast low-bit matmul kernels in Triton
  ☆187Updated last week
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 6 months ago
• Qualcomm-AI-research / gptvq
  ☆27Updated 9 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆58Updated 2 months ago
• Distributed-AI / PipeTransformer
  PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021
  ☆55Updated 3 years ago