Returns a specification to start this module under a supervisor.

See Supervisor.

Copies buffer to device with given device ID.

Returns the name of the default client.

Fetches a client with the given name from configuration.

Sends buffer from device outfeed to the given process tagged by ref.

Note: XLA does not support tuple outfeed shapes. Passing one will simply block the operation indefinitely. Instead, convert the tuple into multiple outfeed operations.

Returns a map of supported platforms with device information.

Sends data_and_shapes to device infeed.

data_and_shapes must be a list of two element tuples where the first element is a binary or a flat list of binaries and the second element is a EXLA.Shape.

Note: XLA does not support tuple infeed shapes when running on host. Passing one will simply block the operation indefinitely. Instead, convert the tuple into multiple infeed operations.