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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
build (3.7, 2.4.*, lib)	1m 12s
build (3.7, 2.4.*, research)	53s
	2m 5s