xnn.common.train

Modules

callbacks	Trainer callbacks: hooks into the training loop (checkpointing, early stopping, progress bar or your own).
loggers	Training loggers: where the trainer's metrics go (console, CSV, TensorBoard or your own).
losses	Weighted energy / force / stress (and dipole / polarizability / charge) loss.
trainer	Training loop: batch training, device selection, validation, checkpointing.