Migrate Flax dataclass to the newest JAX pytree keypath API.#2921
Merged
copybara-service[bot] merged 1 commit intomainfrom Mar 8, 2023
Merged
Migrate Flax dataclass to the newest JAX pytree keypath API.#2921copybara-service[bot] merged 1 commit intomainfrom
copybara-service[bot] merged 1 commit intomainfrom
