Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
PrototypeCommon

kingoflolz/swarm-jax

Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes

Owner
kingoflolz (personal account)
Stars
Stars: 241
Forks
Forks: 22
Language
Python
License
None
Created
2020
Tier
Prototype (100 to 249 ★) What this tier means
View on GitHub

More repos by kingoflolz

All 2 repos by kingoflolz

Nearby in Python

All Python repos