v0.1.4

github-actions released this 15 Mar 20:51

Added

Support for generic transformation function in PyTreeCheckpointHandler.
Support n-digit checkpoint step format.

Fixed

Eliminate Flax dependency to fix circular dependency problem.

Assets 2