awwong1 / torchprof

Related projects ⓘ

Alternatives and complementary repositories for torchprof

• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆571Updated 6 months ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆514Updated 5 years ago
• Santosh-Gupta / SpeedTorch
  Library for faster pinned CPU <-> GPU transfer in Pytorch
  ☆683Updated 4 years ago
• pytorch / contrib
  Implementations of ideas from recent papers
  ☆391Updated 3 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆495Updated 3 years ago
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆360Updated 6 years ago
• 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 …
  ☆194Updated 4 years ago
• JJGO / shrinkbench
  PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.
  ☆424Updated last year
• NVIDIA / runx
  Deep Learning Experiment Management
  ☆639Updated last year
• jacobkimmel / pytorch_modelsize
  Estimates the size of a PyTorch model in memory
  ☆352Updated 4 years ago
• BayesWatch / sequential-imagenet-dataloader
  A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.
  ☆238Updated 3 years ago
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆545Updated 3 years ago
• 1adrianb / pytorch-estimate-flops
  Estimate/count FLOPS for a given neural network using pytorch
  ☆304Updated 2 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,020Updated 3 months ago
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆291Updated 4 years ago
• jacobgil / pytorch-pruning
  PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference
  ☆875Updated 5 years ago
• chester256 / Model-Compression-Papers
  Papers for deep neural network compression and acceleration
  ☆396Updated 3 years ago
• ptrblck / pytorch_misc
  Code snippets created for the PyTorch discussion board
  ☆545Updated 3 years ago
• utsaslab / MONeT
  MONeT framework for reducing memory consumption of DNN training
  ☆173Updated 3 years ago
• google-research / rigl
  End-to-end training of sparse deep neural networks with little-to-no performance loss.
  ☆317Updated last year
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆296Updated 3 years ago
• seba-1511 / dist_tuto.pth
  Official code for "Writing Distributed Applications with PyTorch", PyTorch Tutorial
  ☆255Updated last year
• facebookresearch / kill-the-bits
  Code for: "And the bit goes down: Revisiting the quantization of neural networks"
  ☆636Updated 4 years ago
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆818Updated 3 months ago
• TimDettmers / sparse_learning
  Sparse learning library and sparse momentum resources.
  ☆379Updated 2 years ago
• zsef123 / EfficientNets-PyTorch
  A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."
  ☆309Updated 4 years ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆208Updated this week
• memoiry / Awesome-model-compression-and-acceleration
  ☆662Updated 3 years ago
• enyac-group / single-path-nas
  Single-Path NAS: Designing Hardware-Efficient ConvNets in less than 4 Hours
  ☆396Updated 3 years ago