API Reference EXLA v#0.12.0

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

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

A buffer where data is kept in a binary.

Functions for managing EXLA.Client.

Extension point for lowering selected Nx.block/4 blocks to XLA custom calls (stablehlo.custom_call in StableHLO MLIR).

Result of EXLA.CustomCall.call/4 when lowering a tagged Nx.block/4 to stablehlo.custom_call.

An EXLA DeviceBuffer for data allocated in the device.

Wrapper around XLA's executable.

Combined type and shape information about tensors.