Alternatives and similar repositories for pytorch-benchmark

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

• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆161Updated 2 months ago
• 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.
  ☆131Updated 11 months ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆82Updated 2 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• meta-pytorch / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆223Updated last year
• facebookresearch / bit
  Code repo for the paper BiT Robustly Binarized Multi-distilled Transformer
  ☆112Updated 2 years ago
• meta-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…
  ☆244Updated last month
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆158Updated 2 years ago
• lambdal / deeplearning-benchmark
  Benchmark Suite for Deep Learning
  ☆278Updated last month
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆48Updated 11 months ago
• ShishirPatil / poet
  ML model training for edge devices
  ☆167Updated 2 years ago
• triton-inference-server / pytorch_backend
  The Triton backend for the PyTorch TorchScript models.
  ☆164Updated this week
• Qualcomm-AI-research / transformer-quantization
  ☆207Updated 4 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆163Updated 2 years ago
• cli99 / flops-profiler
  pytorch-profiler
  ☆51Updated 2 years ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆216Updated this week
• ultralytics / thop
  Profile PyTorch models for FLOPs and parameters, helping to evaluate computational efficiency and memory usage.
  ☆65Updated 2 weeks ago
• meta-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…
  ☆181Updated 2 months ago
• IntelLabs / Model-Compression-Research-Package
  A library for researching neural networks compression and acceleration methods.
  ☆140Updated 2 months ago
• DACUS1995 / pytorch-mmap-dataset
  A custom pytorch Dataset extension that provides a faster iteration and better RAM usage
  ☆46Updated last year
• gpuopenanalytics / pynvml
  Provide Python access to the NVML library for GPU diagnostics
  ☆249Updated 2 months ago
• IST-DASLab / OBC
  Code for the NeurIPS 2022 paper "Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning".
  ☆129Updated 2 years ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆121Updated 11 months ago
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆428Updated last week
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆110Updated last year
• NVlabs / EfficientDL
  ☆34Updated 5 months ago
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆148Updated 2 months ago
• MingSun-Tse / Awesome-Efficient-ViT
  Recent Advances on Efficient Vision Transformers
  ☆54Updated 2 years ago
• haochengxi / Train_Transformers_with_INT4
  ☆157Updated 2 years ago
• quic / efficient-transformers
  This library empowers users to seamlessly port pretrained models and checkpoints on the HuggingFace (HF) hub (developed using HF transfor…
  ☆83Updated this week