Implementations of ideas from recent papers
☆392Dec 22, 2020Updated 5 years ago
Alternatives and similar repositories for contrib
Users that are interested in contrib are comparing it to the libraries listed below
Sorting:
- Stochastic Weight Averaging in PyTorch☆977Aug 1, 2021Updated 4 years ago
- torchbearer: A model fitting library for PyTorch☆641Dec 4, 2023Updated 2 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,742Updated this week
- Detectorch - detectron for PyTorch☆558Oct 30, 2018Updated 7 years ago
- PyTorch code for the "Deep Neural Networks with Box Convolutions" paper☆510Jan 20, 2020Updated 6 years ago
- PyTorch layer-by-layer model profiler☆607May 23, 2021Updated 4 years ago
- A lightweight library for PyTorch training tools and utilities☆1,720Feb 10, 2026Updated 2 weeks ago
- Submission to https://pytorch.org/hub/☆1,432Apr 15, 2024Updated last year
- Deal with bad samples in your dataset dynamically, use Transforms as Filters, and more!☆377Sep 22, 2022Updated 3 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆320Aug 4, 2022Updated 3 years ago
- TorchCV: a PyTorch vision library mimics ChainerCV☆396Dec 25, 2018Updated 7 years ago
- Pytorch implementation of CoordConv introduced in 'An intriguing failing of convolutional neural networks and the CoordConv solution' pap…☆402Jul 20, 2018Updated 7 years ago
- lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.☆378Nov 19, 2022Updated 3 years ago
- Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.☆485Feb 12, 2022Updated 4 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,548Jul 31, 2021Updated 4 years ago
- Totally Versatile Miscellanea for Pytorch☆477Mar 20, 2022Updated 3 years ago
- Model analyzer in PyTorch☆1,501Mar 19, 2023Updated 2 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,822Sep 5, 2019Updated 6 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Feb 21, 2020Updated 6 years ago
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs☆1,335Jul 8, 2025Updated 7 months ago
- Neural networks training pipeline based on PyTorch☆312Jun 1, 2020Updated 5 years ago
- PyTorch implementations of algorithms for density estimation☆586May 13, 2021Updated 4 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆870Jun 11, 2020Updated 5 years ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,911Jul 23, 2023Updated 2 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 4 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,990Feb 5, 2026Updated 3 weeks ago
- A domain specific language to express machine learning workloads.☆1,765Apr 28, 2023Updated 2 years ago
- Scaling and Benchmarking Self-Supervised Visual Representation Learning☆587Oct 12, 2021Updated 4 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Mar 2, 2024Updated last year
- AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)☆415Jan 13, 2021Updated 5 years ago
- High-level batteries-included neural network training library for Pytorch☆402Feb 4, 2022Updated 4 years ago
- ICLR Reproducibility Challenge 2019☆220May 23, 2019Updated 6 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,612Jun 16, 2021Updated 4 years ago
- PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"☆2,729Jul 6, 2023Updated 2 years ago
- Python package facilitating the use of Bayesian Deep Learning methods with Variational Inference for PyTorch☆362Oct 12, 2019Updated 6 years ago
- Ape-X DQN & DDPG with pytorch & tensorboard☆102Jun 18, 2019Updated 6 years ago
- Code for https://arxiv.org/abs/1810.04622☆140Aug 28, 2019Updated 6 years ago
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆570Jun 13, 2022Updated 3 years ago