Skip to content

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

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 #2241

Annotations

2 warnings

Python 3.9 on ubuntu-latest jax=newest

succeeded Oct 25, 2024 in 3m 41s