[ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".

A large (+2500) collection of color maps for Python

Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

A compositional diagramming and animation library as an eDSL in Python

⚡ Flashbax: Accelerated Replay Buffers in JAX

Generative Flow Networks - GFlowNet

DynGFN: Bayesian Dynamic Causal Discovery using Generative Flow Networks

TorchCFM: a Conditional Flow Matching library

Laplace approximations for Deep Learning.

Amortized Inference for Causal Structure Learning, NeurIPS 2022

DiBS: Differentiable Bayesian Structure Learning, NeurIPS 2021

TorchOpt is an efficient library for differentiable optimization built upon PyTorch.

GFlowNet library

A curated list of resources about generative flow networks (GFlowNets).

Gaussian processes in JAX.

GFlowNet library specialized for graph & molecular data

RL Environments in JAX 🌍

A general-purpose, deep learning-first library for constrained optimization in PyTorch

Gaussian process modelling in Python

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend

Code for the paper "Learning Differential Equations that are Easy to Solve"

Implementation of For The Win (FTW) agent as introduced by Deepmind.

An offline deep reinforcement learning library

Simple, Elegant, Typed Argument Parsing with argparse

Code for "Data-Efficient Reinforcement Learning with Self-Predictive Representations"