Create a Client — pjrt_client • pjrt

Create a PJRT client for a specific device.

Usage

pjrt_client(platform = NULL)

Arguments

platform: (character(1) | NULL)
Platform name (e.g., "cpu", "cuda", "metal"). If NULL, use PJRT_PLATFORM environment variable or default to "cpu".

Value

PJRTClient

Extractors

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

Examples

pjrt_client("cpu")