API Reference EXLA v0.5.2

modules

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.

Wrapper around XLA's ops.

Wrapper around XLA's shape.