Version 0.7.3
What's Changed
- Fix Documentation Typo by @peterdavidfagan in #3265
- bump version number by @chiamp in #3273
- added cursor api by @chiamp in #3246
- Improve Typing support by @cgarciae in #3242
- feat: add configs for vae example by @SauravMaheshkar in #3254
- fix stackoverflow when loading pickled module by @cgarciae in #3286
- Remove string from checkpointing example and unbreak doctest in head by @IvyZX in #3288
- Improve kw_only_dataclass by @cgarciae in #3293
- updated haiku upgrade guide by @chiamp in #3292
- added logical partitioning to pjit guide by @chiamp in #3290
- Rolling back ba9e24a by @copybara-service in #3295
- Add RNN FLIP by @cgarciae in #2585
- [flax] Add module path to nn.module. by @copybara-service in #3300
- Minor fix to jax_utils.prefect_to_device by @anuragarnab in #3308
- add tf-text to doc requirements by @cgarciae in #3317
- Allow apply's method argument to accept submodules by @cgarciae in #3281
- Update handling of typed PRNG keys by @jakevdp in #3314
- 0.7.3 release by @IvyZX in #3326
New Contributors
- @peterdavidfagan made their first contribution in #3265
- @SauravMaheshkar made their first contribution in #3254
Full Changelog: v0.7.2...v0.7.3