jax

b8b690e5 - Add use_high_dynamic_range_gumbel flag which allows sampling gumbel such

Add use_high_dynamic_range_gumbel flag which allows sampling gumbel such that it more closely matches the CDF for low probably events (less than 2**-nmant). Because -log(-log(x)) is more sensitive close to 1 than 0, we must use -log(-logp1(-x)) instead to make better use of the extra range around 0. PiperOrigin-RevId: 732757388