Skip to content

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

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

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