Allan Zhou

Results 5 repositories owned by Allan Zhou

metalearning-symmetries

58
Stars
8
Forks
58
Watchers

Experiments for Meta-Learning Symmetries by Reparameterization

midGPT

24
Stars
4
Forks
24
Watchers

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

nfn

93
Stars
7
Forks
93
Watchers

NF-Layers for constructing neural functionals.