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

Loading