Add negative sampling as part of the `Loader.collate()` method

[tizianocitro]

Since DataLoader is consumed by MultiModelTrainer (which delegates to Lightning's Trainer.fit()), the epoch loop is managed by Lightning. We need a set_epoch() method that must be called automatically via a Lightning Callback.