Home

ZeRO-transformer

Public

Two implementations of ZeRO-1 optimizer sharding in JAX

Creat:2022-09-07T22:54:42
Update:2025-02-26T11:30:19
14
Stars
0
Stars Increase