jax.Array.device

abstract property Array.device: Device | Sharding

Array API-compatible device attribute.

For single-device arrays, this returns a Device. For sharded arrays, this returns a Sharding.