flax
Fix jax.tree_util.register_dataclass in older JAX versions.
#3885

Merged

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

copybara-service merged 1 commit into main from test_627885708

copybara-service force pushed 2 years ago

Fix jax.tree_util.register_dataclass in older JAX versions.

ea3bcab6

copybara-service force pushed to ea3bcab6 2 years ago

copybara-service merged ea3bcab6 into main 2 years ago

copybara-service deleted the test_627885708 branch 2 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone