Pinned Loading

1. grokking grokking Public

   This repo aims to reproduce "grokking" phenomenon in modular addition task, viz. ( a , b ) ↦ ( a + b ) mod 113 , using a 1-layer transformer model with bidirectional attention.

   Python

2. maml-rl maml-rl Public

   Implementation of Model-Agnostic Meta-Learning (MAML) for Reinforcement Learning using PyTorch and TorchRL.

   Python

3. nano-optax nano-optax Public

   Classic optimizers in JAX.

   Python

4. ott ott Public

   Forked from ott-jax/ott

   Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.

   Python