package nx

  1. Overview
  2. Docs
package nx
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Nx_coreSource

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.