p-wol

gaussian-preact

Computations related to the paper "Gaussian Pre-Activations in Neural Networks: Myth or Reality?"

GroupedNewton

Second-order optimization method adapted to neural networks

deepchaos

Experiments for the paper "Exponential expressivity in deep neural networks through transient chaos"

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

diffusion

Denoising Diffusion Probabilistic Models

ema-pytorch

A simple way to keep track of an Exponential Moving Average (EMA) version of your pytorch model

KFAC-Pytorch

Pytorch implementation of KFAC and E-KFAC (Natural Gradient).

p-wol.github.io

pytorch-fid

Compute FID scores with PyTorch.

pyvarinf

Python package facilitating the use of Bayesian Deep Learning methods with Variational Inference for PyTorch

sgd_ht