lucaslingle / mu_transformerLinks

Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
30Updated 2 weeks ago

Alternatives and similar repositories for mu_transformer

Users that are interested in mu_transformer are comparing it to the libraries listed below

Sorting: