nn Overview

Functional

Neurite's nn.functional (abbreviated as nnf) provides GPU accelerated functional interfaces for working with tensors.

Models

The nn.models module provides a collection of a few pre-defined neural network architectures for various tasks.

Modules

The nn.modules module contains building blocks for constructing neural networks, including layers, loss functions, and optimizers.