Blake Hechtman
a10854786b
[JAX] randint goes from [min, max) so [min - N, max + M] actually maps to [min,
...
max]
PiperOrigin-RevId: 572034040
2023-10-09 14:05:39 -07:00
Jake VanderPlas
b407620d1d
random_test: fix deprecation warnings for key tests
...
Some versions of numpy on some platforms raise warnings when custom PRNG keys
are passed to np.assert_array_equal. Address this by creating a specific function
for comparing key values.
2023-10-06 13:21:23 -07:00
Jake VanderPlas
3d503e01dc
random_test: remove unnecessary test utilities
2023-10-05 15:33:14 -07:00
jax authors
8f911e1512
random_test: Split into two so that each target is small enough to fit within a medium timeout.
...
PiperOrigin-RevId: 571146766
2023-10-05 15:28:51 -07:00
