jax.Array.committed#

propertyArray.committed:bool[source]#

Whether the array is committed or not.

An array is committed when it is explicitly placed on device(s) via JAXAPIs. For example,jax.device_put(np.arange(8), jax.devices()[0]) iscommitted to device 0. Whilejax.device_put(np.arange(8)) is uncommittedand will be placed on the default device.

Computations involving some committed inputs will happen on the committeddevice(s) and the result will be committed on the same device(s).Invoking an operation on arguments that are committed to different device(s)will raise an error.

For example:

`a=jax.device_put(np.arange(8),jax.devices()[0])b=jax.device_put(np.arange(8),jax.devices()[1])a+b #Raisesanerror`

Seehttps://docs.jax.dev/en/latest/faq.html#controlling-data-and-computation-placement-on-devicesfor more information.