google-deepmind/optax v0.0.91 on GitHub

It is the latest version compatible with Python 3.6. See #222 for more details.

Main changes since 0.0.9

Use an explicit type for base classes of optimizer states. #211
Replace optax.OptState with NamedTuple, when it is used as a base class. #218
Set optax.OptState to chex.ArrayTree. #219
Add quick start notebook to Optax. #143
Add Optax regression flax_example.py. by @8bitmp3 in #138
Add scale_gradient. #198
Add a set_to_zero transformation. #202
Add utilities for numerical stability in the public API. #227
Change int64 to safely incremented int32 in wrappers. by @rdaems in #187
Add an example of meta-optimization to the docs. #231

Full Changelog: v0.0.9...v0.0.91