Skip to contents

Convert a platform name to a PJRT client or verify that an object is already a client.

Usage

as_pjrt_client(x)

Arguments

x

(PJRTClient | character(1))
Either a PJRT client object or a platform name (e.g., "cpu", "cuda", "metal").

Value

PJRTClient