View Source API Reference Axon v0.7.0

A high-level interface for creating neural network models.

Activation functions.

Module for rendering various visual representations of Axon models.

Parameter initializers.

Functional implementations of common neural network layer operations.

Abstraction for modeling a reduction of a dataset with an accumulated state for a number of epochs.

Accumulated state in an Axon.Loop.

Implementations of loss-scalers for use in mixed precision training.

Loss functions.

Metric functions.

Utilities for creating mixed precision policies.

Model State Data Structure.

Represents a missing value of an optional node.

Model quantization.

Quantized Layer Implementations.

Representation of a quantized tensor.

Container for returning stateful outputs from Axon layers.