flax
Migrate Flax dataclass to the newest JAX pytree keypath API.
#2921
Merged

Migrate Flax dataclass to the newest JAX pytree keypath API. #2921

copybara-service merged 1 commit into main from test_513588016
copybara-service
copybara-service copybara-service force pushed from aa3761f9 2 years ago
copybara-service copybara-service force pushed 2 years ago
copybara-service copybara-service force pushed 2 years ago
copybara-service copybara-service force pushed 2 years ago
copybara-service copybara-service force pushed 2 years ago
cgarciae
cgarciae commented on 2023-03-06
copybara-service copybara-service force pushed 2 years ago
copybara-service copybara-service changed the title Migrate Flax dataclass to the newest JAX pytree API. Migrate Flax dataclass to the newest JAX pytree keypath API. 2 years ago
copybara-service copybara-service force pushed 2 years ago
copybara-service copybara-service force pushed 2 years ago
codecov-commenter
copybara-service copybara-service force pushed 2 years ago
copybara-service copybara-service force pushed 2 years ago
IvyZX Migrate Flax dataclass to the newest JAX pytree keypath API.
c7fa10a0
copybara-service copybara-service force pushed to c7fa10a0 2 years ago
copybara-service copybara-service merged c7fa10a0 into main 2 years ago
copybara-service copybara-service deleted the test_513588016 branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone