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.3k 247

2. scalax scalax Public

   A simple library for scaling up JAX programs

   Python 113 6

3. mlxu mlxu Public

   Machine Learning eXperiment Utilities

   Python 38 6

4. mintext mintext Public

   Minimal but scalable implementation of large language models in JAX

   Python 12

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 8

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.3k 371