Skip to content

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

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

Annotations

1 warning

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