API Reference EXLA v0.3.0
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.
Wrapper around XLA's ops.
Wrapper around XLA's shape.