kingoflolz / mesh-transformer-jax

Model parallel transformers in JAX and Haiku
6,298Updated last year

Related projects

Alternatives and complementary repositories for mesh-transformer-jax