Giters

google / flax

Flax is a neural network library for JAX that is designed for flexibility.

Home Page:https://flax.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Update documentation to `optax`

andsteing opened this issue · comments

commented

Examples are currently being refactored to use optax #1356

This issue tracks updating the documentation (readthedocs, pydoc) to mark flax.optim as deprecated, and to introduce new documentation for using optax.