GitHub Action pytest

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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
test	15s
test_jft_active_learning	18s
test_jft_batchensemble	17s
test_jft_deep_ensemble	22s
test_jft_deterministic	27s
test_jft_heteroscedastic	27s
test_jft_sngp	18s
test_jft_utils	19s
	2m 43s