memorax.loggers

Logging utilities for tracking training progress.

Core

Logger - Protocol that all loggers implement.

MultiLogger - Composite logger that dispatches to multiple backends.

Backends

DashboardLogger - Rich terminal dashboard.

FileLogger - Logs to file.

WandbLogger - Weights & Biases integration.

TensorBoardLogger - TensorBoard integration.

CheckpointLogger - Orbax checkpoint saving.