Implementations of ideas from recent papers
☆391Dec 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
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,752Mar 11, 2026Updated last week
- torchbearer: A model fitting library for PyTorch☆641Dec 4, 2023Updated 2 years ago
- Detectorch - detectron for PyTorch☆559Oct 30, 2018Updated 7 years ago
- PyTorch code for the "Deep Neural Networks with Box Convolutions" paper☆509Jan 20, 2020Updated 6 years ago
- A lightweight library for PyTorch training tools and utilities☆1,719Mar 6, 2026Updated 2 weeks ago
- PyTorch layer-by-layer model profiler☆606May 23, 2021Updated 4 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆320Aug 4, 2022Updated 3 years 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!☆378Sep 22, 2022Updated 3 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
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Jul 31, 2021Updated 4 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
- TorchCV: a PyTorch vision library mimics ChainerCV☆396Dec 25, 2018Updated 7 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,819Sep 5, 2019Updated 6 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,989Feb 5, 2026Updated last month
- Totally Versatile Miscellanea for Pytorch☆477Mar 20, 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,628Mar 25, 2022Updated 3 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Feb 21, 2020Updated 6 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 4 years ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,909Jul 23, 2023Updated 2 years ago
- A domain specific language to express machine learning workloads.☆1,764Apr 28, 2023Updated 2 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆869Jun 11, 2020Updated 5 years ago
- Model analyzer in PyTorch☆1,501Mar 19, 2023Updated 3 years ago
- AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)☆415Jan 13, 2021Updated 5 years ago
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs☆1,334Jul 8, 2025Updated 8 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,063Mar 2, 2024Updated 2 years 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
- ☆28Jul 20, 2020Updated 5 years ago
- Bayesian optimization in PyTorch☆3,481Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,936Updated this week
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,610Jun 16, 2021Updated 4 years ago
- Scaling and Benchmarking Self-Supervised Visual Representation Learning☆587Oct 12, 2021Updated 4 years ago
- Experiments with differentiable stacks and queues in PyTorch☆145Oct 7, 2019Updated 6 years ago
- Repository has been moved: https://github.com/adobe/antialiased-cnns☆164Sep 16, 2021Updated 4 years ago
- A mix of GAN implementations including progressive growing☆1,628Oct 12, 2021Updated 4 years ago
- PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"☆2,726Jul 6, 2023Updated 2 years ago