Repository info

optax

• Summary: a summary of optax
• Description: Optax is a gradient processing and optimization library for JAX.
• Stars: 1937
• Number of forks: 246