๐ŸŽฑ API Reference#

algorithms

Efficiency methods for training.

callbacks

Callbacks that run at each training loop Event.

core

Central components used by other modules.

datasets

Natively supported datasets.

functional

Functional API for applying algorithms in your own training loop.

loggers

Loggers to store metrics and artifacts.

models

The models module contains the ComposerModel base class along with reference implementations of many common models.

optim

Optimizers and learning rate schedulers.

profiler

Performance profiling tools.

trainer

Train models with flexible insertion of algorithms.

utils

Helper utilities.