flax
83eebe27 - Avoid assert_array_equal for PRNG keys.

Commit

1 year ago

Avoid assert_array_equal for PRNG keys. This will soon error due to https://github.com/jax-ml/jax/pull/24481 PiperOrigin-RevId: 694612853

References

Author

Jake VanderPlas

Committer

a-googler

Parents