Pinned Loading

1. babel babel Public

   Minimal LLM in Jax.

   Python

2. mu_transformer mu_transformer Public

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

   Python 32 1

3. pytorch_rl2 pytorch_rl2 Public

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

   Python 72 11

4. draw_tfp draw_tfp Public

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

   Python

5. metavb metavb Public

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

   Python

6. letterbox_generator letterbox_generator Public

   Generate NYT-style 'Letter Boxed' games!

   Python 3 1