package nx

You can search for identifiers within the package.

in-package search v0.2.0

Module `Nx_core`Source

Core modules for nx.

This module re-exports core building blocks used by backends and the high-level Nx frontend.

Sourcemodule Dtype : sig ... end

Tensor element dtypes.

Sourcemodule Shape : sig ... end

Concrete shape operations.

Sourcemodule View : sig ... end

Strided tensor views.

Sourcemodule Backend_intf : sig ... end

Backend interface used by frontend functors.

Sourcemodule Rng : sig ... end

RNG key utilities.

Sourcemodule Make_frontend (B : Backend_intf.S) : sig ... end

Frontend functor parameterized by a backend implementation.