Devices

Get the addressable devices for a PJRT client.

Usage

devices(client = NULL)

Arguments

client

(PJRTClient)
Object convertible to a PJRTClient.

Value

list of PJRTDevice

Examples

# Create client (defaults to CPU)
client <- pjrt_client()
devices(client)
#> [[1]]
#> <CpuDevice(id=0)>
#>