flax
[JAX] Replace uses of deprecated `jax.ops.index_update(x, idx, y)` APIs with their up-to-date, more succinct equivalent `x.at[idx].set(y)`.
#1594

Merged

[JAX] Replace uses of deprecated `jax.ops.index_update(x, idx, y)` APIs with their up-to-date, more succinct equivalent `x.at[idx].set(y)`. #1594

copybara-service merged 1 commit into main from test_400203457

google-cla added cla: yes

levskaya approved these changes on 2021-10-04

copybara-service force pushed from 28fae03e 4 years ago

[JAX] Replace uses of deprecated `jax.ops.index_update(x, idx, y)` AP…

facbc5d7

copybara-service force pushed to facbc5d7 4 years ago

copybara-service merged facbc5d7 into main 4 years ago

copybara-service deleted the test_400203457 branch 4 years ago

Reviewers

levskaya

Assignees

No one assigned

Labels

cla: yes

Milestone

No milestone