v1.4.4: Split out dev dependencies and revert jax.tree.map for backwards compatibility on older jax versions

MichaelTMatthews released this 17 Oct 12:30

· 24 commits to main since this release

e1c69ca

v1.4.4

Assets 2