python-optax 0.1.5
Propagated dependencies: python-absl-py@1.4.0 python-chex@0.1.8 python-jax@0.4.20 python-jaxlib@0.4.20 python-numpy@1.24.4
Channel: guix-science
Home page: https://github.com/google-deepmind/optax/
Licenses: ASL 2.0
Synopsis: Gradient processing and optimization library for JAX
Description:
Optax is a gradient processing and optimization library for JAX.
Total results: 1