Contains functions and settings related to backend specifications.

Represents a backend for dfx tensors

Defines a new op implementing a binary function and its derivatives. Instances of this class are used with the Tensor.Op method to define a new differentiable tensor function that supports forward, reverse, and nested differentiation.

Defines a new op implementing an elementwise binary function and its derivatives. Instances of this class are used with the Tensor.Op method to define a new differentiable tensor function that supports forward, reverse, and nested differentiation.

Contains functions and settings related to device specifications.

Represents a device specification.

Represents the type of a device.

Tensor operations

Contains functions and settings related to tensor element types

Represents a storage type for elements of a tensor

Contains global functions and settings related to tensor element types, used when writing backends.

Contains auto-opened utilities related to the dfx programming model.

Contains functions and settings related to print options.

Represents a scalar in the dfx programming model

Contains functions and values related to tensor shapes.

Represents the shape of a tensor.

Represents a multi-dimensional data type containing elements of a single data type.

Defines a new op implementing a unary function and its derivatives. Instances of this class are used with the Tensor.Op method to define a new differentiable tensor function that supports forward, reverse, and nested differentiation.

Defines a new op implementing an elementwise unary function and its derivatives. Instances of this class are used with the Tensor.Op method to define a new differentiable tensor function that supports forward, reverse, and nested differentiation.