mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
17,764 Commits 504 Branches 263 Tags
Commit Graph

3 Commits

Author SHA1 Message Date
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