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