evanatyourservice / llm-jaxLinks

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

Alternatives and similar repositories for llm-jax

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

Sorting: