Alternatives and similar repositories for torch-scan:

Users that are interested in torch-scan are comparing it to the libraries listed below

• 1adrianb / pytorch-estimate-flops
  Estimate/count FLOPS for a given neural network using pytorch
  ☆304Updated 2 years ago
• lukemelas / do-you-even-need-attention
  Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)
  ☆483Updated 3 years ago
• google-research / receptive_field
  Compute receptive fields of your favorite convnets
  ☆437Updated 3 years ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆341Updated 4 years ago
• pabloppp / pytorch-tools
  Useful PyTorch functions and modules that are not implemented in PyTorch by default
  ☆187Updated 9 months ago
• Fangyh09 / pytorch-receptive-field
  Compute CNN receptive field size in pytorch in one line
  ☆357Updated 9 months ago
• antoinebrl / torchextractor
  Feature extraction made simple with torchextractor
  ☆100Updated 3 years ago
• PhoenixDL / rising
  Provides everything needed for high performance data loading and augmentation in pytorch.
  ☆318Updated last year
• bonlime / pytorch-tools
  Tool box for PyTorch
  ☆187Updated 10 months ago
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago
• awwong1 / torchprof
  PyTorch layer-by-layer model profiler
  ☆606Updated 3 years ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆596Updated 9 months 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 …
  ☆197Updated 5 years ago
• shelfwise / receptivefield
  Gradient based receptive field estimation for Convolutional Neural Networks
  ☆338Updated 5 years ago
• ryul99 / pytorch-project-template
  Deep Learning project template for PyTorch (multi-gpu training is supported)
  ☆135Updated last year
• jacobkimmel / pytorch_modelsize
  Estimates the size of a PyTorch model in memory
  ☆356Updated 4 years ago
• fkodom / fft-conv-pytorch
  Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
  ☆490Updated last year
• PistonY / torch-toolbox
  🛠 Toolbox to extend PyTorch functionalities
  ☆419Updated 9 months ago
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆210Updated 2 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆501Updated 3 years ago
• benjs / nfnets_pytorch
  Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".
  ☆159Updated 3 years ago
• vra / flopth
  A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.
  ☆125Updated 2 months ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆471Updated 2 years ago
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆292Updated 4 years ago
• digantamisra98 / EvoNorm
  Unofficial PyTorch Implementation of EvoNorm
  ☆121Updated 3 years ago
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆187Updated last year
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆364Updated 6 years ago
• michaelsdr / momentumnet
  Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities
  ☆209Updated 9 months ago
• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆421Updated 4 months ago
• lessw2020 / Best-Deep-Learning-Optimizers
  Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable
  ☆211Updated 3 years ago