Skip to content
build

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu"). #566

Sign in to view logs

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu"). #566

Run time

Learn about OS pricing on GitHub Actions
Job Run time
build (3.7, 2.4.*, lib)
56s
build (3.7, 2.4.*, research)
1m 2s
1m 58s