API Reference Nx v#0.11.0

Numerical Elixir.

The behaviour for tensor backends.

Creates a batch of tensors (and containers).

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

Common constants used in computations.

A protocol that teaches defn how to traverse data structures.

Numerical functions.

The specification and helper functions for custom defn compilers.

Functions to deal with composite data types.

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

The expression used by Nx.Defn.Compiler.

A module for splitting Nx.Defn.Expr into stages.

All imported functionality available inside defn blocks.

A defn token used by hooks.

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

Functions for loading and dumping floating-point formats not supported by the Erlang VM.

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

Converts a data structure to a container lazily.

Nx conveniences for linear algebra.

A mesh is a named collection of devices arranged in a logical shape.

Represents a reference to a value in memory.

Pseudo-random number generators.

Serving encapsulates client and server work to perform batched requests.

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

The tensor struct and the behaviour for backends.

Testing functions for Nx tensor assertions.

Conveniences for working with types.