Lightning

WeightsLab inside a LightningModule, replacing the training loop with Lightning’s trainer while keeping full per-sample observability.