flax
Fix PRNG handling in `nn.jit` under `nn.scan`.
#4359

Merged

Fix PRNG handling in `nn.jit` under `nn.scan`. #4359

copybara-service merged 1 commit into main from test_692231015

copybara-service force pushed 1 year ago

copybara-service changed the title ~~Fix PRNG handling of `nn.jit` under `nn.scan`.~~ Fix PRNG handling in `nn.jit` under `nn.scan`. 1 year ago

copybara-service force pushed 1 year ago

Fix PRNG handling in `nn.jit` under `nn.scan`.

0f631a27

copybara-service force pushed to 0f631a27 1 year ago

copybara-service merged 0f631a27 into main 1 year ago

copybara-service deleted the test_692231015 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone