v1.0.0

Breaking API changes.

v0.1.2

More general pack/unpack for serialization helper

v0.1.1

Unofficial support for batch axes for various ops

Release v0.1.0

Naming convention nit

Release v0.0.4

Fix typing issue for setup.py

Release v0.0.3

Add flax serialization support, bump version

Release v0.0.2

README nit

Release v0.0.1

Support Python 3.6, 3.7; add publish action