A stateful pytree library for training neural networks.
-
Updated
Aug 31, 2025 - Python
#
opax
Here is 1 public repository matching this topic...
Add this topic to your repo
To associate your repository with the opax topic, visit your repo's landing page and select "manage topics."