support passing `jax.random.PRNGKey` inputs in jax #2075

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Check the code format	4m 14s
Run tests (tensorflow)	8m 40s
Run tests (jax)	8m 9s
Run tests (torch)	4m 46s
Run tests (numpy)	3m 56s
	29m 45s