Popular repositories Loading

1. assets assets Public

2. t5x t5x Public

   Forked from google-research/t5x

   Python

3. flax flax Public

   Forked from google/flax

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

   Python 1

4. paxml paxml Public

   Forked from google/paxml

   Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…

   Python 1

5. praxis praxis Public

   Forked from google/praxis

   Python 1

6. maxtext maxtext Public

   Forked from AI-Hypercomputer/maxtext

   A simple, performant and scalable Jax LLM!

   Python