Alternatives and similar repositories for torchprof:

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

• yaysummeriscoming / DALI_pytorch_demo
  Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …
  ☆197Updated 5 years ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆599Updated 9 months ago
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆366Updated 6 years ago
• NVIDIA / runx
  Deep Learning Experiment Management
  ☆639Updated 2 years ago
• pytorch / contrib
  Implementations of ideas from recent papers
  ☆391Updated 4 years ago
• Santosh-Gupta / SpeedTorch
  Library for faster pinned CPU <-> GPU transfer in Pytorch
  ☆685Updated 5 years ago
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆292Updated 4 years ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆513Updated 6 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆501Updated 3 years ago
• JJGO / shrinkbench
  PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.
  ☆428Updated last year
• TimDettmers / sparse_learning
  Sparse learning library and sparse momentum resources.
  ☆379Updated 2 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,045Updated 6 months ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆298Updated 3 years ago
• ptrblck / pytorch_misc
  Code snippets created for the PyTorch discussion board
  ☆557Updated 4 years ago
• zsef123 / EfficientNets-PyTorch
  A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."
  ☆309Updated 5 years ago
• jacobkimmel / pytorch_modelsize
  Estimates the size of a PyTorch model in memory
  ☆357Updated 4 years ago
• karanchahal / distiller
  A large scale study of Knowledge Distillation.
  ☆219Updated 4 years ago
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆546Updated 4 years ago
• 1adrianb / pytorch-estimate-flops
  Estimate/count FLOPS for a given neural network using pytorch
  ☆304Updated 2 years ago
• Lyken17 / Efficient-PyTorch
  My best practice of training large dataset using PyTorch.
  ☆1,092Updated 9 months ago
• joe-siyuan-qiao / WeightStandardization
  Standardizing weights to accelerate micro-batch training
  ☆547Updated 3 years ago
• enyac-group / single-path-nas
  Single-Path NAS: Designing Hardware-Efficient ConvNets in less than 4 Hours
  ☆396Updated 4 years ago
• facebookresearch / kill-the-bits
  Code for: "And the bit goes down: Revisiting the quantization of neural networks"
  ☆634Updated 4 years ago
• eladhoffer / convNet.pytorch
  ConvNet training using pytorch
  ☆345Updated 4 years ago
• BayesWatch / sequential-imagenet-dataloader
  A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.
  ☆238Updated 3 years ago
• chester256 / Model-Compression-Papers
  Papers for deep neural network compression and acceleration
  ☆396Updated 3 years ago
• NVlabs / Taylor_pruning
  Pruning Neural Networks with Taylor criterion in Pytorch
  ☆315Updated 5 years ago
• pytorch / tvm
  TVM integration into PyTorch
  ☆452Updated 5 years ago
• synxlin / nn-compression
  A Pytorch implementation of Neural Network Compression (pruning, deep compression, channel pruning)
  ☆155Updated 7 months ago
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆253Updated 2 years ago