Skip to content

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

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

Annotations

2 warnings

Python 3.11 on ubuntu-latest jax=newest

succeeded Oct 28, 2024 in 1m 45s