Allow unfrozen dataclasses
Created by: PhilipVinc
This is the only commit for ode4jax in https://github.com/PhilipVinc/netket_dynamics to work.
It's just an internal flag to control whever dataclasses are frozen or not. I use it in ode4jax
to avoid writing repetitive code.
It should really go in it's own package, but since it's here, and it's a 3 liner.