Unittests & Auto-publish

Replace deprecated `jax.tree_` functions with `jax.tree.` #206

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
pytest-job	1m 42s
publish-job	0s
	1m 42s