PyTorch implementation of Learning to learn by gradient descent by gradient descent.
python main.py
PyTorch implementations of normalizing flow and its variants.
PyNIF3D is an open-source PyTorch-based library for research on neural implicit functions (NIF)-based 3D geometry representation. It aims to accelerate research by providing a modular design that all
geoopt Manifold aware pytorch.optim. Unofficial implementation for “Riemannian Adaptive Optimization Methods” ICLR2019 and more. Installation Make sur
model_based_energy_constrained_compression Code for paper "Energy-Constrained Compression for Deep Neural Networks via Weighted Sparse Projection and
TODO: update this README! Fast Discounted Cumulative Sums in PyTorch This repository implements an efficient parallel algorithm for the computation of
Opacus is a library that enables training PyTorch models with differential privacy. It supports training with minimal code changes required on the cli
EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis This repo contains the official implementations of EigenDamage: Structured Prunin
Tacotron 2 (without wavenet) PyTorch implementation of Natural TTS Synthesis By Conditioning Wavenet On Mel Spectrogram Predictions. This implementati
Code release for "Bayesian Compression for Deep Learning" In "Bayesian Compression for Deep Learning" we adopt a Bayesian view for the compression of
PyTorch misc Collection of code snippets I've written for the PyTorch discussion board. All scripts were testes using the PyTorch 1.0 preview and torc
OptNet: Differentiable Optimization as a Layer in Neural Networks This repository is by Brandon Amos and J. Zico Kolter and contains the PyTorch sourc
PyTorch Implementation of Differentiable ODE Solvers This library provides ordinary differential equation (ODE) solvers implemented in PyTorch. Backpr
This is an differentiable pytorch implementation of SIFT patch descriptor. It is very slow for describing one patch, but quite fast for batch. It can
Tez: a simple pytorch trainer NOTE: Currently, we are not accepting any pull requests! All PRs will be closed. If you want a feature or something does
A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-precision, and PyTorch extensions.
Use Jax functions in Pytorch with DLPack
TorchShard is a lightweight engine for slicing a PyTorch tensor into parallel shards. It can reduce GPU memory and scale up the training when the model has massive linear layers (e.g., ViT, BERT and
PyTorch Sparse This package consists of a small extension library of optimized sparse matrix operations with autograd support. This package currently
Fast and Easy-to-use Distributed Graph Learning for PyTorch Geometric
MentisOculi Pytorch Path Tracer A very simple and small path tracer written in pytorch meant to be run on the GPU Why use pytorch and not some other c
55 Dec 01, 2022
96 Nov 28, 2022
642 Jan 03, 2023
16 Jun 15, 2022
7 Feb 17, 2022
1.3k Dec 29, 2022
107 Apr 20, 2022
4.1k Jan 03, 2023
190 Dec 30, 2022
461 Dec 26, 2022
428 Dec 24, 2022
4.4k Jan 04, 2023
150 Dec 24, 2022
1.1k Jan 04, 2023
56 Sep 13, 2022
106 Dec 17, 2022
275 Nov 22, 2022
757 Jan 04, 2023
221 Dec 22, 2022
222 Dec 01, 2022