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)
  ☆64Updated last year
• BradMcDanel / sdgp
  ☆10Updated 3 years ago
• thu-ml / Jetfire-INT8Training
  ☆51Updated 11 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆167Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• albanD / subclass_zoo
  ☆167Updated last year
• 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 11 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆100Updated 3 weeks ago
• cchan / tccl
  extensible collectives library in triton
  ☆86Updated 2 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆125Updated 7 months ago
• yaozhewei / HAP
  ☆43Updated last year
• spcl / sten
  Sparsity support for PyTorch
  ☆35Updated 3 months ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆157Updated last year
• stanford-futuredata / stk
  ☆105Updated 10 months ago
• Dao-AILab / quack
  A Quirky Assortment of CuTe Kernels
  ☆117Updated this week
• pytorch-labs / superblock
  A block oriented training approach for inference time optimization.
  ☆33Updated 10 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆132Updated 2 months ago
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆170Updated this week
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆77Updated last week
• HazyResearch / fly
  ☆208Updated 2 years ago
• osayamenja / Kleos
  Complete GPU residency for ML.
  ☆17Updated last week
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆224Updated 10 months ago
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆131Updated last week
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆56Updated 3 years ago
• LiuXiaoxuanPKU / GACT-ICML
  ☆42Updated 2 years ago
• xvyaward / owq
  Code for the AAAI 2024 Oral paper "OWQ: Outlier-Aware Weight Quantization for Efficient Fine-Tuning and Inference of Large Language Model…
  ☆63Updated last year
• Qualcomm-AI-research / gptvq
  ☆31Updated last year
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆39Updated 2 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆132Updated 3 years ago
• Jokeren / triton-samples
  ☆28Updated 5 months ago