GitHub - kingoflolz/mesh-transformer-jax: Model parallel transformers in JAX and Haiku
Model parallel transformers in JAX and Haiku. Contribute to kingoflolz/mesh-transformer-jax development by creating an account on GitHub.
Source: github.com
Stay updated with breaking news from Mesh Transformer. Get real-time updates on events, politics, business, and more. Visit us for reliable news and exclusive interviews.
Model parallel transformers in JAX and Haiku. Contribute to kingoflolz/mesh-transformer-jax development by creating an account on GitHub.