Rare find

mesh-transformer-jax. Model parallel transformers in JAX and Haiku

github.com/kingoflolz/mesh-transformer-jax
by @kingoflolz

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.