Bayesian learning and inference for state space models

torchview: visualize pytorch models

Pytorch implementation of lfads, and hierarchical extension

Models and examples built with TensorFlow

Pytorch implementation of Block Neural Autoregressive Flow

A library for differentiable nonlinear optimization

Tutorial on amortized optimization for learning to optimize over continuous domains

A collection of modern/faster/saner alternatives to common unix commands.

Awesome resources on normalizing flows.

Normalizing flows in PyTorch

Neural Spline Flow, RealNVP, Autoregressive Flow, 1x1Conv in PyTorch.

PyTorch implementation of normalizing flow models

Pytorch implementations of density estimation algorithms: BNAF, Glow, MAF, RealNVP, planar flows

Learning to draw samples: with application to amortized maximum likelihood estimator for generative adversarial learning

A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN

Recurrent continuous reinforcement learning algorithms implemented in Pytorch.

Toolbox of models, callbacks, and datasets for AI/ML researchers.

Sampling based model predictive control (MPC) using MuJoCo simulator

Reinforcement learning algorithms for MuJoCo tasks

Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)

ConvLSTM/ConvGRU (Encoder-Decoder) with PyTorch on Moving-MNIST

A library that uses hardware acceleration to load sequences of video frames to facilitate machine learning training

VS Code in the browser

Latex code for making neural networks diagrams

A template for learning-based projects with managed, reproducible experiments

Efficient Householder Transformation in PyTorch

Differentiation through cone programs

Matrix square root with gradient support for PyTorch

Differentiable convex optimization layers