flax
Avoid passing non-boolean mask to `where` argument of `jax.numpy` reductions. Non-boolean mask inputs have been deprecated for several releases, and will result in an error starting in JAX v0.8.0.
#4923

Merged

Avoid passing non-boolean mask to `where` argument of `jax.numpy` reductions. Non-boolean mask inputs have been deprecated for several releases, and will result in an error starting in JAX v0.8.0. #4923

copybara-service merged 1 commit into main from test_802622436

copybara-service force pushed from 1e576acd to 5834ea5b 207 days ago

copybara-service force pushed from 5834ea5b to c896e736 207 days ago

copybara-service force pushed from c896e736 to d4bd3846 206 days ago

Avoid passing non-boolean mask to `where` argument of `jax.numpy` red…

4887f7dc

copybara-service force pushed from d4bd3846 to 4887f7dc 206 days ago

copybara-service closed this 206 days ago

copybara-service deleted the test_802622436 branch 206 days ago

copybara-service merged 4887f7dc into main 206 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone