enerzyme.tasks.lightning_utils# Classes CollectOutputCallback() EMACallback(use_ema[, ema_decay, ...]) LightningModel(model, loss_terms, dump_dir, ...) MonitorCallback(monitor)