Alternatives and similar repositories for pytorch-benchmark:

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

• 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 4 months ago
• 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…
  ☆229Updated 2 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆107Updated this week
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆155Updated 3 months ago
• fumihwh / onnx-pytorch
  A code generator from ONNX to PyTorch code
  ☆135Updated 2 years ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆84Updated last year
• Qualcomm-AI-research / transformer-quantization
  ☆202Updated 3 years ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆222Updated 8 months ago
• alenic / timm-models-explorer
  Timm model explorer
  ☆37Updated 11 months ago
• yqhu / profiler-workshop
  Example code for profiler workshop
  ☆34Updated 3 years ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated last year
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆354Updated this week
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆157Updated last year
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆137Updated last month
• facebookresearch / bit
  Code repo for the paper BiT Robustly Binarized Multi-distilled Transformer
  ☆104Updated last year
• IntelLabs / DyNAS-T
  Dynamic Neural Architecture Search Toolkit
  ☆29Updated 3 months ago
• facebookresearch / DeltaCNN
  DeltaCNN End-to-End CNN Inference of Sparse Frame Differences in Videos
  ☆60Updated 2 years ago
• triton-inference-server / pytorch_backend
  The Triton backend for the PyTorch TorchScript models.
  ☆144Updated 3 weeks ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆160Updated 2 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆115Updated 7 months ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆189Updated this week
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆154Updated 10 months ago
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆39Updated last month
• elliotwaite / pytorch-hooks-tutorial
  Examples of using PyTorch hooks, as covered in my YouTube tutorial video.
  ☆33Updated last year
• IST-DASLab / OBC
  Code for the NeurIPS 2022 paper "Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning".
  ☆113Updated last year
• onnx / neural-compressor
  Model compression for ONNX
  ☆88Updated 4 months ago
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆191Updated last year
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆37Updated 8 months ago
• mlcommons / logging
  MLPerf™ logging library
  ☆33Updated 3 weeks ago
• IntelLabs / FP8-Emulation-Toolkit
  PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.
  ☆107Updated 3 months ago