jax
6fdf64d7 - add back an OptState that is a proper pytree

Commit

6 years ago

add back an OptState that is a proper pytree

References

#521 - remove JaxTuples from optimizer states

Author

mattjj

mattjj

Parents

Loading