Skip to content

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
test_562017973
Open

Replace deprecated jax.linear_util.wrap_init with jax.extend.linear_util.wrap_init.#252
copybara-service[bot] wants to merge 1 commit intomainfrom
test_562017973

Conversation

@copybara-service
Copy link

Replace deprecated jax.linear_util.wrap_init with jax.extend.linear_util.wrap_init.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant