API Reference EXLA v0.9.2

Modules

EXLA

Google's XLA (Accelerated Linear Algebra) compiler/backend for Nx.

EXLA.Backend

A Nx tensor backend for the data kept on the device.

EXLA.BinaryBuffer

A buffer where data is kept in a binary.

EXLA.Client

Functions for managing EXLA.Client.

EXLA.DeviceBuffer

An EXLA DeviceBuffer for data allocated in the device.

EXLA.Executable

Wrapper around XLA's executable.

EXLA.Typespec

Combined type and shape information about tensors.