Alternatives and similar repositories for torchprofile

Users that are interested in torchprofile are comparing it to the libraries listed below

• 1adrianb / pytorch-estimate-flops
  Estimate/count FLOPS for a given neural network using pytorch
  ☆306Updated 3 years ago
• ucbrise / actnn
  ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
  ☆199Updated 2 years ago
• JJGO / shrinkbench
  PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.
  ☆431Updated 2 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆508Updated 4 years ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆519Updated 6 years ago
• Tiiiger / QPyTorch
  Low Precision Arithmetic Simulation in PyTorch
  ☆286Updated last year
• awwong1 / torchprof
  PyTorch layer-by-layer model profiler
  ☆608Updated 4 years ago
• aojunzz / NM-sparsity
  ☆243Updated 3 years ago
• amirgholami / ZeroQ
  [CVPR'20] ZeroQ: A Novel Zero Shot Quantization Framework
  ☆280Updated last year
• mit-han-lab / hardware-aware-transformers
  [ACL'20] HAT: Hardware-Aware Transformers for Efficient Natural Language Processing
  ☆336Updated last year
• idstcv / ZenNAS
  ☆228Updated 4 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆163Updated 2 years ago
• MingSun-Tse / Efficient-Deep-Learning
  Collection of recent methods on (deep) neural network compression and acceleration.
  ☆951Updated 7 months ago
• NVlabs / Taylor_pruning
  Pruning Neural Networks with Taylor criterion in Pytorch
  ☆319Updated 6 years ago
• facebookresearch / AttentiveNAS
  code for "AttentiveNAS Improving Neural Architecture Search via Attentive Sampling"
  ☆105Updated 4 years ago
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆204Updated 2 years ago
• JiahuiYu / slimmable_networks
  Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019
  ☆924Updated 2 years ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆301Updated 5 months ago
• csyhhu / Awesome-Deep-Neural-Network-Compression
  Summary, Code for Deep Neural Network Quantization
  ☆557Updated 5 months ago
• cedrickchee / awesome-ml-model-compression
  Awesome machine learning model compression research papers, quantization, tools, and learning material.
  ☆537Updated last year
• yhhhli / BRECQ
  Pytorch implementation of BRECQ, ICLR 2021
  ☆285Updated 4 years ago
• Zhen-Dong / HAWQ
  Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.
  ☆451Updated 2 years ago
• mit-han-lab / amc
  [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices
  ☆445Updated last year
• VITA-Group / TENAS
  [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …
  ☆168Updated 3 years ago
• walkerning / aw_nas
  aw_nas: A Modularized and Extensible NAS Framework
  ☆251Updated 2 years ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆222Updated 7 months ago
• WoosukKwon / retraining-free-pruning
  [NeurIPS 2022] A Fast Post-Training Pruning Framework for Transformers
  ☆192Updated 2 years ago
• yhhhli / APoT_Quantization
  PyTorch implementation for the APoT quantization (ICLR 2020)
  ☆280Updated 11 months ago
• microsoft / Tutel
  Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4
  ☆938Updated this week
• marcoancona / TorchPruner
  On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for n…
  ☆167Updated 5 years ago