Alternatives and similar repositories for pytorch-benchmark:

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

• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆221Updated 6 months ago
• Qualcomm-AI-research / transformer-quantization
  ☆198Updated 3 years ago
• facebookresearch / DepthShrinker
  [ICML 2022] "DepthShrinker: A New Compression Paradigm Towards Boosting Real-Hardware Efficiency of Compact Neural Networks", by Yonggan …
  ☆71Updated 2 years ago
• facebookresearch / DeltaCNN
  DeltaCNN End-to-End CNN Inference of Sparse Frame Differences in Videos
  ☆60Updated last year
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆84Updated last year
• facebookresearch / bit
  Code repo for the paper BiT Robustly Binarized Multi-distilled Transformer
  ☆103Updated last year
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆154Updated 2 months ago
• IntelLabs / DyNAS-T
  Dynamic Neural Architecture Search Toolkit
  ☆29Updated 2 months ago
• NVlabs / EfficientDL
  ☆31Updated 8 months ago
• sony / nnabla-nas
  Neural Architecture Search for Neural Network Libraries
  ☆58Updated last year
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆157Updated last year
• 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.
  ☆125Updated 2 months ago
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆67Updated 8 months ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆160Updated 2 years ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆89Updated this week
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆115Updated 6 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated 11 months ago
• IntelLabs / Model-Compression-Research-Package
  A library for researching neural networks compression and acceleration methods.
  ☆140Updated 5 months ago
• jaeho-lee / layer-adaptive-sparsity
  In progress.
  ☆63Updated 10 months ago
• pytorch-labs / superblock
  A block oriented training approach for inference time optimization.
  ☆32Updated 6 months ago
• alenic / timm-models-explorer
  Timm model explorer
  ☆37Updated 10 months ago
• Qualcomm-AI-research / FP8-quantization
  ☆136Updated last year
• andravin / spio
  Efficient CUDA kernels for training convolutional neural networks with PyTorch.
  ☆39Updated 3 weeks ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆46Updated 2 months ago
• ziplab / EcoFormer
  [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"
  ☆70Updated 2 years ago
• IST-DASLab / OBC
  Code for the NeurIPS 2022 paper "Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning".
  ☆109Updated last year
• VITA-Group / Structure-LTH
  [ICML 2022] "Coarsening the Granularity: Towards Structurally Sparse Lottery Tickets" by Tianlong Chen, Xuxi Chen, Xiaolong Ma, Yanzhi Wa…
  ☆32Updated last year
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆187Updated this week
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆257Updated 2 years ago
• pytorch / multipy
  torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…
  ☆178Updated 2 months ago