1 Commits

Author	SHA1	Message	Date
Suraj Patil	44f5b260fe	flax model parallel training (#12590) ... * update scripts * add copyright * add logging * cleanup * add z loss * add readme * shard description * update readme	2021-07-14 22:55:44 +05:30