Build and minimal test

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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
build (3.8)	25s
build (3.9)	43s
build (3.10)	41s
	1m 49s