Skip to content

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

[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"). #569

Annotations

1 error and 1 warning

The logs for this run have expired and are no longer available.