API Reference Nx v0.7.1

Modules

Nx

Numerical Elixir.

Nx.Backend

The behaviour for tensor backends.

Nx.Batch

Creates a batch of tensors (and containers).

Nx.BinaryBackend

An opaque backend written in pure Elixir that stores the data in Elixir's binaries.

Nx.Constants

Common constants used in computations.

Nx.Container

A protocol that teaches defn how to traverse data structures.

Nx.Defn

Numerical functions.

Nx.Defn.Compiler

The specification and helper functions for custom defn compilers.

Nx.Defn.Composite

Functions to deal with composite data types.

Nx.Defn.Evaluator

The default implementation of a Nx.Defn.Compiler that evaluates the expression tree against the tensor backend.

Nx.Defn.Expr
Nx.Defn.Kernel

All imported functionality available inside defn blocks.

Nx.Defn.Token

A defn token used by hooks.

Nx.Defn.Tree

Helper functions to traverse defn expressions, either as single nodes or in-depth.

Nx.Heatmap

Provides a heatmap that is printed using ANSI colors in the terminal.

Nx.LazyContainer

Converts a data structure to a container lazily.

Nx.LinAlg

Nx conveniences for linear algebra.

Nx.LinAlg.Cholesky
Nx.LinAlg.Eigh
Nx.LinAlg.QR
Nx.Random

Pseudo-random number generators.

Nx.Serving

Serving encapsulates client and server work to perform batched requests.

Nx.Stream

The protocol for streaming data in and out of backends.

Nx.TemplateBackend

An opaque backend written that is used as template to declare the type, shape, and names of tensors to be expected in the future.

Nx.Tensor

The tensor struct and the behaviour for backends.

Nx.Type

Conveniences for working with types.