Skip to content

Use jax.tree_util.tree_map in place of deprecated tree_multimap.

48f1833
Select commit
Loading
Failed to load commit list.
Open

Use jax.tree_util.tree_map in place of deprecated tree_multimap. #208

Use jax.tree_util.tree_map in place of deprecated tree_multimap.
48f1833
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs