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
  ☆200Updated 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
  ☆505Updated 4 years ago
• awwong1 / torchprof
  PyTorch layer-by-layer model profiler
  ☆605Updated 4 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆162Updated 2 years ago
• MingSun-Tse / Efficient-Deep-Learning
  Collection of recent methods on (deep) neural network compression and acceleration.
  ☆952Updated 5 months ago
• aojunzz / NM-sparsity
  ☆240Updated 2 years ago
• Tiiiger / QPyTorch
  Low Precision Arithmetic Simulation in PyTorch
  ☆284Updated last year
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆520Updated 6 years ago
• cedrickchee / awesome-ml-model-compression
  Awesome machine learning model compression research papers, quantization, tools, and learning material.
  ☆535Updated 11 months ago
• idstcv / ZenNAS
  ☆227Updated 3 years ago
• amirgholami / ZeroQ
  [CVPR'20] ZeroQ: A Novel Zero Shot Quantization Framework
  ☆281Updated last year
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆300Updated 3 months ago
• mit-han-lab / hardware-aware-transformers
  [ACL'20] HAT: Hardware-Aware Transformers for Efficient Natural Language Processing
  ☆336Updated last year
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆291Updated 4 years ago
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆200Updated 2 years ago
• NVlabs / Taylor_pruning
  Pruning Neural Networks with Taylor criterion in Pytorch
  ☆319Updated 5 years ago
• JiahuiYu / slimmable_networks
  Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019
  ☆924Updated 2 years ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆221Updated 5 months ago
• mit-han-lab / amc
  [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices
  ☆445Updated last year
• csyhhu / Awesome-Deep-Neural-Network-Compression
  Summary, Code for Deep Neural Network Quantization
  ☆552Updated 3 months ago
• Zhen-Dong / HAWQ
  Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.
  ☆446Updated 2 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,069Updated last week
• microsoft / Tutel
  Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4
  ☆913Updated this week
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆854Updated last year
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆366Updated 7 years ago
• yhhhli / BRECQ
  Pytorch implementation of BRECQ, ICLR 2021
  ☆282Updated 4 years ago
• chester256 / Model-Compression-Papers
  Papers for deep neural network compression and acceleration
  ☆401Updated 4 years ago
• jshilong / FisherPruning
  Group Fisher Pruning for Practical Network Compression(ICML2021)
  ☆160Updated 2 years ago