jax
8309836d - Merge pull request #821 from georgedahl/optimizer_state_saving

Commit

6 years ago

Merge pull request #821 from georgedahl/optimizer_state_saving Changes to jax optimizers.py to facilitate serialization of optimizer state.

References

Author

mattjj

Parents