0.5.0

What's Changed

• 📝 Update pull_request_template.md by @alvarobartt in #25
• 🍱 Add support for fsspec file-systems by @alvarobartt in #23
• 🧪 Add assertions over pytrees with chex by @alvarobartt in #27
• 🐛 Fix issue when restoring a VarCollection by @alvarobartt in #28

Full Changelog: 0.4.0...0.5.0

0.4.0

What's Changed

• ♻️ Refactor on how safejax is structured by @alvarobartt in #21

Full Changelog: 0.3.0...0.4.0

0.3.0

What's Changed

• ✨ safejax support for objax model params by @alvarobartt in #10
• ✏️ Add DictionaryLike type in safejax.typing by @alvarobartt in #13
• 📝 Add mkdocs and mkdocs-material documentation by @alvarobartt in #14
• 🔒 Lock dependencies with pip-tools by @alvarobartt in #15
• 📝 Update README.md and docs/usage.md by @alvarobartt in #16

Full Changelog: 0.2.0...0.3.0

0.2.0

What's Changed

• ⚡️ Overall improvements, bug fixes, more unit tests, and dm-haiku compatibility tested by @alvarobartt in #6

Full Changelog: 0.1.1...0.2.0

0.1.1

⛔️ safejax 0.1.1 is not recommended as it has a bug when unflattening some dictionaries, so please use 0.2.0 instead.

What's Changed

• 🚚 Rename frozen_or_unfrozen_dict to params in serialize by @alvarobartt in #2
• ⬆️ Upgrade safejax to 0.1.1 by @alvarobartt in #4
• ✨ Add benchmarks/resnet50.py by @alvarobartt in #5
• 🧪 Add more unit test cases e.g. ResNet FrozenDict by @alvarobartt in #3

Full Changelog: 0.1.0...0.1.1

0.1.0

⛔️ safejax 0.1.0 is not recommended as it has a bug when unflattening some dictionaries, so please use 0.2.0 instead.

What's Changed

• ✨ Add Flax-serialization with safetensors for FrozenDict by @alvarobartt in #1

New Contributors

• @alvarobartt made their first contribution in #1

Full Changelog: https://github.com/alvarobartt/safejax/commits/0.1.0