Popular repositories Loading

1. vanilla_transformer_from_scratch_with_JAX vanilla_transformer_from_scratch_with_JAX Public

   Jupyter Notebook 10 1

2. EasyLM EasyLM Public

   Forked from young-geng/EasyLM

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

   Python 1

3. flax flax Public

   Forked from google/flax

   Flax is a neural network library for JAX that is designed for flexibility.

   Python

4. RetNet_JAX_LM RetNet_JAX_LM Public

   Python

5. mamba mamba Public

   Forked from state-spaces/mamba

   Python

6. gpt-fast gpt-fast Public

   Forked from pytorch-labs/gpt-fast

   Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

   Python