Fix jax.tree_util.register_dataclass in older JAX versions. #3885
Google CLA / cla/google
succeeded
Apr 25, 2024 in 2s
✅ All contributors are covered under a CLA with Google
See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).
ℹ️ Googlers: Go here to view more details and manage scans for this pull request.
Details
Loading