Optax

github.comtool

A gradient processing and optimization library for JAX.

from

Data Science

added

2026-10-10

likes

0

Data Science › Deep Learning > JAX: “A gradient processing and optimization library for JAX.”