Implementation of PSGD optimizer in JAX
-
Updated
Dec 31, 2024 - Python
Implementation of PSGD optimizer in JAX
Add a description, image, and links to the psgd topic page so that developers can more easily learn about it.
To associate your repository with the psgd topic, visit your repo's landing page and select "manage topics."