Cluster 653410

2 repos

Python · 2

Related clusters

salesforce/jaxformer

Minimal library to train LLMs on TPU in JAX with pjit().

Python

300

33 commits

kingoflolz/mesh-transformer-jax

Model parallel transformers in JAX and Haiku

Python

6,382

143 commits