Pinned Loading

1. EasyLM EasyLM Public

   Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

   Python 2.4k 256

2. scalax scalax Public

   A simple library for scaling up JAX programs

   Python 127 10

3. mintext mintext Public

   Minimal but scalable implementation of large language models in JAX

   Python 26

4. mlxu mlxu Public

   Machine Learning eXperiment Utilities

   Python 45 6

5. tpu_pod_commander tpu_pod_commander Public

   TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.

   Python 17

6. openlm-research/open_llama openlm-research/open_llama Public

   OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset

   7.4k 379