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

2 warnings

Cancel Previous Runs

succeeded Apr 25, 2024 in 4s