API Reference EXLA v0.7.2

Modules

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.

Wrapper around XLA's builder.

Functions for managing EXLA.Client.

Wrapper around XLA's computation.

An EXLA DeviceBuffer for data allocated in the device.

Wrapper around XLA's executable.

High-level operations built on top of EXLA.Op.

Representation of an MLIR Function or func.func type.

Representation of an MLIR module.

Representation of an MLIR Value.

Wrapper around XLA's ops.

Wrapper around XLA's shape.