extra credit Search 85,062 technologies ⌘K Download for Mac

optax

A gradient processing and optimization library in JAX.

Category
Python packages
pip install optax
↑↓ move ↵ open