Pinned

1. mu_transformer mu_transformer Public

   Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.

   Python 19 1

2. e-lra e-lra Public

   Streamlined variant of Long-Range Arena with pinned dependencies, automated data downloads, and deterministic shuffling.

   Python 2

3. pytorch_rl2 pytorch_rl2 Public

   Implementation of 'RL^2: Fast Reinforcement Learning via Slow Reinforcement Learning'

   Python 48 10

4. pytorch_drl pytorch_drl Public

   Deep reinforcement learning algorithms implemented in Pytorch.

   Python

5. draw_tfp draw_tfp Public

   Tensorflow Implementation of 'DRAW: A Recurrent Neural Network For Image Generation'

   Python

6. metavb metavb Public

   Official implementation of 'Meta-Learning with Variational Bayes'.

   Python