Commit b2e98d6
committed
jax.tree_util.register_pytree_node
1 parent 0e74913 commit b2e98d6
File tree
3 files changed
+155
-174
lines changed- src/array_api_extra
- _lib/_utils
- tests
3 files changed
+155
-174
lines changed1 parent 0e74913 commit b2e98d6
0 commit comments