flax
[nnx] fix Rngs dtype check
#4820
Merged

Loading