Module Nx_core

Core modules for nx.

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

module Dtype : sig ... end

Tensor element dtypes.

module Shape : sig ... end

Concrete shape operations.

module View : sig ... end

Strided tensor views.

module Backend_intf : sig ... end

Backend interface used by frontend functors.

module Rng : sig ... end

RNG key utilities.

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

Frontend functor parameterized by a backend implementation.