v1.0.8
- custom callback functions ([pre|loss|pos]_callback) have been refactored to be more flexible.
- Supported input arguments include: ["inputs", "outputs", "params", "opt", "seq", "aux", "key"].
- The pre_callback function can be used to modify inputs before prediction, loss_callback to add custom loss.
- adding support for Optax optimizers
Full Changelog: v1.0.7...v1.0.8