Skip to content
acme-tests

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backe… #1182

Sign in to view logs

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backe…

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backe… #1182

Run time

Learn about OS pricing on GitHub Actions
Job Run time
run-acme-tests (python:3.8)
2m 8s
run-acme-tests (python:3.9)
2m 19s
4m 27s