t5x: remove references to deprecated jax.random.KeyArray #1408
test_569549521% was force-pushed and no longer has any new commits.
Pushing new commits will allow the pull request to be re-opened.
Pushing new commits will allow the pull request to be re-opened.