EasyLM

github.comtool

LLMs made easy: Pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   flaxmodels github.com

   tool category #jax link ↗

   Pretrained models for Jax/Flax.

2. ▲
   Lorax github.com

   tool category #jax link ↗

   Automatically apply LoRA to JAX models (Flax, Haiku, etc.)

   
3. ▲
   Deep Learning tutorials with JAX+Flax by Phillip Lippe uvadlc-notebooks.readthedocs.io

   site category #jax link ↗

   A series of notebooks explaining various deep learning concepts, from basics (e.g. intro to JAX/Flax, activiation functions) to recent advances (e.g., Vision Transformers, SimCLR), with translations to PyTorch.

4. ▲
   FLAX github.com

   tool category #python-data-science link ↗

   A neural network library for JAX that is designed for flexibility.

5. ▲
   Train LLM From Scratch github.com

   tool category #datascience link ↗

   A straightforward method for training your LLM, from downloading data to generating text.

   
6. ▲
   FedJAX github.com

   tool category #jax link ↗

   Federated learning in JAX, built on Optax and Haiku.

   

JAX › Libraries: “LLMs made easy: Pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.”