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.