Avoid assert_array_equal for PRNG keys.#4363

copybara-service[bot] merged 1 commit intomaingoogle/flax:mainfrom

test_694551516google/flax:test_694551516Copy head branch name to clipboard