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

Merged

copybara-service[bot] merged 1 commit intomainfrom

Nov 8, 2024

Commits on Nov 8, 2024

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

danielsuo
authored and
Flax Authors
committed