Resolve "make jax_tools slightly less annoying: non-breaking versions"

Closes #28 (closed). Serialization checks put in backlog, probably better served w a rewrite which doesnt have so much state.