Update jax.tree_util.tree_map to jax.tree.map #3387

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
lint (3.9)	4m 1s
test-modeling (3.9)	23m 8s
test-inference (3.9)	27m 43s
examples (3.9)	10m 39s
	1h 5m 31s