Skip to content

Add handling for JAX typed PRNG keys in chex.assert_trees_all_equal #2240

Add handling for JAX typed PRNG keys in chex.assert_trees_all_equal

Add handling for JAX typed PRNG keys in chex.assert_trees_all_equal #2240

Annotations

2 warnings

Python 3.9 on ubuntu-latest jax=0.4.27

succeeded Oct 25, 2024 in 3m 42s