kingoflolz / swarm-jaxLinks

Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
239Updated 2 years ago

Alternatives and similar repositories for swarm-jax

Users that are interested in swarm-jax are comparing it to the libraries listed below

Sorting: