Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
The recent partitionable Threefry upgrade affects binomial sampling under the RBG PRNG scheme because the implementation of `jax.random.binomial` derives internal subkeys with a call to `split`. This led a randomized test to fail by pushing its numeric closeness check just beyond its current relative tolerance. This is very likely a false failure, so we update the rtol. PiperOrigin-RevId: 723100174
- Loading branch information