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.
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.