Popular repositories Loading

1. DeepExplorationEnsembling DeepExplorationEnsembling Public

   A distributed DQN implementation in JAX/Haiku with randomized priors and Stein variational Gradient Descent

   Python 2

2. ModelBasedRLTensorDecomp ModelBasedRLTensorDecomp Public

   Code for my position paper https://arxiv.org/abs/2110.14524#

   Python

3. SmalldataMNIST SmalldataMNIST Public

   A short experiment showcasing the use of stein variational gradient descent https://arxiv.org/abs/2106.11642 in ensembles to quantify uncertainty in neural network classifiers

   Python

4. VqvaePixelSNAIL VqvaePixelSNAIL Public

   An implementation of PixelSNAIL https://arxiv.org/abs/1712.09763 in JAX/Haiku to learn a prior over the latent space as encoded by a VQVAE https://arxiv.org/abs/1711.00937

   Python

5. blackjax blackjax Public

   Forked from blackjax-devs/blackjax

   BlackJAX is a sampling library designed for ease of use, speed and modularity.

   Python

6. gymnax gymnax Public

   Forked from RobertTLange/gymnax

   RL Environments in JAX 🌍

   Python