evanatyourservice / llm-jaxLinks

Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
19Updated 2 weeks ago

Alternatives and similar repositories for llm-jax

Users that are interested in llm-jax are comparing it to the libraries listed below

Sorting: