Replace deprecated jax.linear_util.wrap_init with jax.extend.linear_util.wrap_init.#252
Open
copybara-service[bot] wants to merge 1 commit intomainfrom
Open
Replace deprecated jax.linear_util.wrap_init with jax.extend.linear_util.wrap_init.#252copybara-service[bot] wants to merge 1 commit intomainfrom
copybara-service[bot] wants to merge 1 commit intomainfrom
Commits
Commits on Sep 1, 2023
- authored andcommitted
