Devices — devices • pjrt

Get the addressable devices.

Usage

devices(x = NULL, ...)

Arguments

x: An object to get devices from: a PJRTClient, a PJRTLoadedExecutable, or NULL (default client).
...: Additional arguments (currently unused).

Value

list of PJRTDevice

Examples

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