jax
defb53f4 - Document jax.tree_util.register_dataclass

Commit

1 year ago

Document jax.tree_util.register_dataclass

References

#21268 - Document jax.tree_util.register_dataclass

Author

jakevdp

jakevdp

Parents

Loading