Pinned Loading

1. Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.

   Python 24 4

2. NF-Layers for constructing neural functionals.

   Python 93 7

3. Experiments for Meta-Learning Symmetries by Reparameterization

   Python 58 8