flax
Migrate from `jax.experimental.enable_x64` to `jax.enable_x64`.
#5011

Merged

Migrate from `jax.experimental.enable_x64` to `jax.enable_x64`. #5011

copybara-service merged 1 commit into main from test_816838669

copybara-service force pushed from 44942194 to 7f54ce25 104 days ago

copybara-service force pushed from 7f54ce25 to 1c68d7a4 104 days ago

copybara-service force pushed from 1c68d7a4 to 6d217a21 103 days ago

Migrate from `jax.experimental.enable_x64` to `jax.enable_x64`.

1453d295

copybara-service force pushed from 6d217a21 to 1453d295 103 days ago

copybara-service merged 1453d295 into main 103 days ago

copybara-service deleted the test_816838669 branch 103 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone