Create a PJRT Device — pjrt_device • pjrt

Create a PJRT Device from an R object.

Usage

pjrt_device(device)

Arguments

device: (any)
The device.

Value

PJRTDevice

Extractors

platform() for a character(1) representation of the platform.

Examples

# Show available devices for CPU client
devices(pjrt_client("cpu"))
#> [[1]]
#> <CpuDevice(id=0)>
#> 
# Create device 0 for CPU client
dev <- pjrt_device("cpu:0")
dev
#> <CpuDevice(id=0)>