Templatization of the DigiLoader
Templatization of FairMQSampler