Repository info

optax

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