Use orbax for saving trained models #320
Description
https://github.com/google/orbax
Maybe this allows use to remove some code? Also related to #169.
{{ message }}
https://github.com/google/orbax
Maybe this allows use to remove some code? Also related to #169.