Pinned

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.2k 233

2. 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.2k 366

3. scalax scalax Public

   A simple library for scaling up JAX programs

   Python 105 6

4. mlxu mlxu Public

   Machine Learning eXperiment Utilities

   Python 37 6

5. m3ae_public m3ae_public Public

   Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation

   Python 93 12