What's Changed * 📝 Update `pull_request_template.md` by alvarobartt in https://github.com/alvarobartt/safejax/pull/25 * 🍱 Add support for `fsspec` file-systems by alvarobartt in https://github.com/alvarobartt/safejax/pull/23 * 🧪 Add assertions over `pytrees` with `chex` by alvarobartt in https://github.com/alvarobartt/safejax/pull/27 * 🐛 Fix issue when restoring a `VarCollection` by alvarobartt in https://github.com/alvarobartt/safejax/pull/28
What's Changed * ✨ `safejax` support for `objax` model params by alvarobartt in https://github.com/alvarobartt/safejax/pull/10 * ✏️ Add `DictionaryLike` type in `safejax.typing` by alvarobartt in https://github.com/alvarobartt/safejax/pull/13 * 📝 Add `mkdocs` and `mkdocs-material` documentation by alvarobartt in https://github.com/alvarobartt/safejax/pull/14 * 🔒 Lock dependencies with `pip-tools` by alvarobartt in https://github.com/alvarobartt/safejax/pull/15 * 📝 Update `README.md` and `docs/usage.md` by alvarobartt in https://github.com/alvarobartt/safejax/pull/16
What's Changed * ⚡️ Overall improvements, bug fixes, more unit tests, and `dm-haiku` compatibility tested by alvarobartt in https://github.com/alvarobartt/safejax/pull/6
What's Changed * 🚚 Rename `frozen_or_unfrozen_dict` to `params` in `serialize` by alvarobartt in https://github.com/alvarobartt/safejax/pull/2 * ⬆️ Upgrade `safejax` to 0.1.1 by alvarobartt in https://github.com/alvarobartt/safejax/pull/4 * ✨ Add `benchmarks/resnet50.py` by alvarobartt in https://github.com/alvarobartt/safejax/pull/5 * 🧪 Add more unit test cases e.g. ResNet FrozenDict by alvarobartt in https://github.com/alvarobartt/safejax/pull/3