Jestimator Versions Save

Amos optimizer with JEstimator lib.

v0.3.3

1 year ago
  • Python 3.7 compatibility.
  • Add d_coef and c_coef to Amos hyper-parameter.
  • Support for flax_mutables in checkpointing.
  • Bug fix data/pipeline_rec and simplified data_utils code.