Geeks
fattorib
/
ZeRO-transformer
Two implementations of ZeRO-1 optimizer sharding in JAX
Primary Language
Python
MIT License
MIT
ZeRO-transformer GitHub
Markdown
Readme
6
Issues
13
Stargazers
2
Watchers
Stargazers
akashgokul
AmrMKayid
co:X
arjunbazinga
@woven-planet
BigBalloon8
Edinburgh
boom90lb
San Diego, CA
danielz02
Massachusetts Institute of Technology
eitanturok
@databricks
exclamaforte
@pytorch
imoneoi
knight9114
Mddct
nschlemm
coder nostra GmbH
ray-ng