Skip to contents

Create a PJRT plugin for a specific platform.

Usage

pjrt_plugin(platform)

Arguments

platform

(character(1))
Platform name (e.g., "cpu", "cuda", "metal").

Value

PJRTPlugin

Extractors