Skip to content

Fix jax.tree_util.register_dataclass in older JAX versions. #12551

Fix jax.tree_util.register_dataclass in older JAX versions.

Fix jax.tree_util.register_dataclass in older JAX versions. #12551

Annotations

1 warning

Run Tests (3.11, doctest)

succeeded Apr 25, 2024 in 2m 4s