Skip to content
ci

Change deprecated jax.tree_util.tree_map to jax.tree.map. Fix argument passed to jax.numpy.finfo call. #183

Sign in to view logs

Change deprecated jax.tree_util.tree_map to jax.tree.map. Fix argument passed to jax.numpy.finfo call.

Change deprecated jax.tree_util.tree_map to jax.tree.map. Fix argument passed to jax.numpy.finfo call. #183

Run time

Learn about OS pricing on GitHub Actions
Job Run time
Python 3.10 on ubuntu-latest
1m 49s
Python 3.11 on ubuntu-latest
1m 51s
Python 3.12 on ubuntu-latest
1m 12s
4m 52s