API Reference Axon v0.1.0

modules

Modules

A high-level interface for creating neural network models.

Activation functions.

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.

Loss functions.

Metric functions.

Utilities for creating mixed precision policies.

Implementations of common gradient-based optimization algorithms.

Functional implementations of common recurrent neural network routines.

Parameter Schedules.

Container for returning stateful outputs from Axon layers.

Parameter update methods.