What's Changed
- Add a link to mctx-az. by @copybara-service[bot] in #90
- Replace deprecated
jax.tree_*functions withjax.tree.*by @copybara-service[bot] in #91 - Fix typo in tree.py by @KazukiOhta in #92
- Change deprecated jax.tree_util.tree_map to jax.tree.map. Fix argument passed to jax.numpy.finfo call. by @carlosgmartin in #95
- Ignore pylint refactoring messages. by @copybara-service[bot] in #101
New Contributors
- @KazukiOhta made their first contribution in #92
Full Changelog: v0.0.5...v0.0.6