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

Commit
6 years ago
add back an OptState that is a proper pytree
Author
Parents
Loading