#
xmap
Here are 2 public repositories matching this topic...
Two implementations of ZeRO-1 optimizer sharding in JAX
-
Updated
Jun 11, 2023 - Python
Improve this page
Add a description, image, and links to the xmap topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the xmap topic, visit your repo's landing page and select "manage topics."