jax-ml / jax-llm-examplesLinks

Minimal yet performant LLM examples in pure JAX
148Updated this week

Alternatives and similar repositories for jax-llm-examples

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

Sorting: