google-deepmind/optax v0.1.6

Optax 0.1.6

on GitHub

What's Changed

• feat: Add support for is_leaf to tree_map_params, so that tree_map_params works with optax.masked.
• feat: migrate to pyproject.toml for building package by @SauravMaheshkar in #513
• fix(README): small typo in README by @jeertmans in #495
• Add support for PolyLoss (ICLR 2022) by @acforvs in #467

New Contributors

• @jeertmans made their first contribution in #495

Full Changelog: v0.1.5...v0.1.6