build

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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
build (3.7, 2.4.*, lib)	1m 11s
build (3.7, 2.4.*, research)	55s
	2m 6s