lucaslingle / mu_transformerLinks

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

Alternatives and similar repositories for mu_transformer

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

Sorting: