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.9, mypy)

succeeded Apr 25, 2024 in 56s