Skip to content

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

Fix jax.tree_util.register_dataclass in older JAX versions.

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

Annotations

1 warning

Test import standalone (3.9)

succeeded Apr 25, 2024 in 30s