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
  ☆305Updated 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.
  ☆430Updated 2 years ago
• awwong1 / torchprof
  PyTorch layer-by-layer model profiler
  ☆607Updated 4 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆503Updated 4 years ago
• aojunzz / NM-sparsity
  ☆236Updated 2 years ago
• idstcv / ZenNAS
  ☆226Updated 3 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆162Updated 2 years ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆517Updated 6 years ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆299Updated last month
• mit-han-lab / hardware-aware-transformers
  [ACL'20] HAT: Hardware-Aware Transformers for Efficient Natural Language Processing
  ☆335Updated last year
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆194Updated 2 years ago
• NVlabs / Taylor_pruning
  Pruning Neural Networks with Taylor criterion in Pytorch
  ☆319Updated 5 years ago
• amirgholami / ZeroQ
  [CVPR'20] ZeroQ: A Novel Zero Shot Quantization Framework
  ☆280Updated last year
• Tiiiger / QPyTorch
  Low Precision Arithmetic Simulation in PyTorch
  ☆279Updated last year
• csyhhu / Awesome-Deep-Neural-Network-Compression
  Summary, Code for Deep Neural Network Quantization
  ☆549Updated last month
• JiahuiYu / slimmable_networks
  Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019
  ☆922Updated 2 years ago
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆291Updated 4 years ago
• yhhhli / BRECQ
  Pytorch implementation of BRECQ, ICLR 2021
  ☆276Updated 3 years ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆218Updated 3 months ago
• MingSun-Tse / Efficient-Deep-Learning
  Collection of recent methods on (deep) neural network compression and acceleration.
  ☆948Updated 3 months ago
• microsoft / Tutel
  Tutel MoE: Optimized Mixture-of-Experts Library, Support DeepSeek FP8/FP4
  ☆851Updated this week
• mit-han-lab / amc
  [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices
  ☆443Updated last year
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆366Updated 7 years ago
• VITA-Group / TENAS
  [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …
  ☆169Updated 3 years ago
• google-research / vmoe
  ☆654Updated 3 weeks ago
• marcoancona / TorchPruner
  On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for n…
  ☆164Updated 5 years ago
• cedrickchee / awesome-ml-model-compression
  Awesome machine learning model compression research papers, quantization, tools, and learning material.
  ☆526Updated 9 months ago
• Zhen-Dong / HAWQ
  Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.
  ☆442Updated 2 years ago
• facebookresearch / AttentiveNAS
  code for "AttentiveNAS Improving Neural Architecture Search via Attentive Sampling"
  ☆104Updated 3 years ago