search

juliuskunze / jaxnet

Concise deep learning for JAX
Apache License 2.0
184 stars 14 forks source link

Allow storing optimizer state #12

Closed juliuskunze closed 4 years ago

juliuskunze commented 4 years ago

Allow save(state, path) according to design laid out in https://github.com/google/jax/issues/1278.