Alternatives and similar repositories for pytorch-benchmark

Users that are interested in pytorch-benchmark are comparing it to the libraries listed below

• IST-DASLab / OBC
  Code for the NeurIPS 2022 paper "Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning".
  ☆120Updated last year
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆64Updated last year
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆223Updated 10 months ago
• Qualcomm-AI-research / transformer-quantization
  ☆204Updated 3 years ago
• IntelLabs / DyNAS-T
  Dynamic Neural Architecture Search Toolkit
  ☆30Updated 6 months ago
• IntelLabs / Model-Compression-Research-Package
  A library for researching neural networks compression and acceleration methods.
  ☆139Updated 9 months ago
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆157Updated 6 months ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆157Updated last year
• Qualcomm-AI-research / FP8-quantization
  ☆149Updated 2 years ago
• VITA-Group / Structure-LTH
  [ICML 2022] "Coarsening the Granularity: Towards Structurally Sparse Lottery Tickets" by Tianlong Chen, Xuxi Chen, Xiaolong Ma, Yanzhi Wa…
  ☆33Updated 2 years ago
• IntelLabs / FP8-Emulation-Toolkit
  PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.
  ☆110Updated 6 months ago
• facebookresearch / DeltaCNN
  DeltaCNN End-to-End CNN Inference of Sparse Frame Differences in Videos
  ☆59Updated 2 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆162Updated 2 years ago
• facebookresearch / bit
  Code repo for the paper BiT Robustly Binarized Multi-distilled Transformer
  ☆108Updated last year
• facebookresearch / DepthShrinker
  [ICML 2022] "DepthShrinker: A New Compression Paradigm Towards Boosting Real-Hardware Efficiency of Compact Neural Networks", by Yonggan …
  ☆71Updated 2 years ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆127Updated this week
• vra / flopth
  A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.
  ☆129Updated 6 months ago
• hahnyuan / PB-LLM
  PB-LLM: Partially Binarized Large Language Models
  ☆152Updated last year
• pytorch / torcheval
  A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…
  ☆233Updated 4 months ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆195Updated last year
• yaozhewei / HAP
  ☆43Updated last year
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆41Updated 3 months ago
• yqhu / profiler-workshop
  Example code for profiler workshop
  ☆33Updated 3 years ago
• Qualcomm-AI-research / gptvq
  ☆30Updated last year
• Guangxuan-Xiao / torch-int
  This repository contains integer operators on GPUs for PyTorch.
  ☆205Updated last year
• Qualcomm-AI-research / pruning-vs-quantization
  ☆22Updated last year
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆194Updated 2 years ago
• thu-ml / Jetfire-INT8Training
  ☆49Updated 10 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆125Updated 2 months ago
• cli99 / flops-profiler
  pytorch-profiler
  ☆51Updated 2 years ago