PyTorch implementation of Learning to learn by gradient descent by gradient descent.
python main.py
AdaBound An optimizer that trains as fast as Adam and as good as SGD, for developing state-of-the-art deep learning models on a wide variety of popula
AccSGD This is the code associated with Accelerated SGD algorithm used in the paper On the insufficiency of existing momentum schemes for Stochastic O
EfficientNet PyTorch Quickstart Install with pip install efficientnet_pytorch and load a pretrained EfficientNet with: from efficientnet_pytorch impor
Use Jax functions in Pytorch with DLPack
Tacotron 2 (without wavenet) PyTorch implementation of Natural TTS Synthesis By Conditioning Wavenet On Mel Spectrogram Predictions. This implementati
News SRU++, a new SRU variant, is released. [tech report] [blog] The experimental code and SRU++ implementation are available on the dev branch which
Here is Poutyne. Poutyne is a simplified framework for PyTorch and handles much of the boilerplating code needed to train neural networks. Use Poutyne
Example implementation of the L0 regularization method described at Learning Sparse Neural Networks through L0 regularization, Christos Louizos, Max W
S3-plugin is a high performance PyTorch dataset library to efficiently access datasets stored in S3 buckets.
Opacus is a library that enables training PyTorch models with differential privacy. It supports training with minimal code changes required on the cli
(Generic) EfficientNets for PyTorch A 'generic' implementation of EfficientNet, MixNet, MobileNetV3, etc. that covers most of the compute/parameter ef
PyTorch to TensorFlow Lite converter
Fast and Easy-to-use Distributed Graph Learning for PyTorch Geometric
PyTorch-LBFGS: A PyTorch Implementation of L-BFGS Authors: Hao-Jun Michael Shi (Northwestern University) and Dheevatsa Mudigere (Facebook) What is it?
A Closer Look at Structured Pruning for Neural Network Compression Code used to reproduce experiments in https://arxiv.org/abs/1810.04622. To prune, w
GPU-accelerated PyTorch implementation of Zero-shot User Intent Detection via Capsule Neural Networks This repository implements a capsule model Inten
Torch-Struct: Structured Prediction Library A library of tested, GPU implementations of core structured prediction algorithms for deep learning applic
3D-RETR: End-to-End Single and Multi-View 3D Reconstruction with Transformers (BMVC 2021) Zai Shi*, Zhao Meng*, Yiran Xing, Yunpu Ma, Roger Wattenhofe
Kaldi-compatible feature extraction with PyTorch, supporting CUDA, batch processing, chunk processing, and autograd
PyTorch Lightning Optical Flow models, scripts, and pretrained weights.
2.9k Dec 27, 2022
205 Jan 02, 2023
7.2k Jan 06, 2023
106 Dec 17, 2022
4.1k Jan 03, 2023
2.1k Jan 01, 2023
534 Dec 17, 2022
202 Nov 10, 2022
138 Jan 03, 2023
1.3k Dec 29, 2022
1.5k Jan 01, 2023
140 Dec 13, 2022
221 Dec 22, 2022
478 Dec 27, 2022
140 Dec 05, 2022
15 Dec 24, 2022
1.1k Jan 07, 2023
36 Dec 21, 2022
119 Jan 03, 2023
105 Dec 16, 2022