Popular repositories Loading

1. mesh-transformer-jax mesh-transformer-jax Public

   Forked from kingoflolz/mesh-transformer-jax

   Model parallel transformers in JAX and Haiku

   Python

2. gpt-neo gpt-neo Public

   Forked from EleutherAI/gpt-neo

   An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.

   Python

3. MTJ-on-TPU_driver0.2 MTJ-on-TPU_driver0.2 Public

   Python

4. mesh-transformer-jax2 mesh-transformer-jax2 Public

   test for v2

   Python

5. mesh-transformer-jax3 mesh-transformer-jax3 Public

   test for v3

   Python

6. mesh-transformer-jax4 mesh-transformer-jax4 Public

   test for v3 jax0.3.25

   Python