support passing `jax.random.PRNGKey` inputs in jax (#651) #2090

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Check the code format	3m 36s
Run tests (tensorflow)	7m 30s
Run tests (jax)	6m 38s
Run tests (torch)	4m 1s
Run tests (numpy)	4m 27s
	26m 12s