You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
It seems that there is currently no open source implementation of optimizer state sharding (ZeRO) in jax. This would be a great addition that greatly simplifies training large models using Adam or Adamw.
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
It seems that there is currently no open source implementation of optimizer state sharding (ZeRO) in jax. This would be a great addition that greatly simplifies training large models using Adam or Adamw.
Beta Was this translation helpful? Give feedback.
All reactions