Cited Paper #1 implementation with imagenet baseline

[zabor2432]

We need to provide pytorch implementation for this paper. Core alogirthm is written in MPI, we need to make this code callable as pytorch lightning callback

[zabor2432]

Remember to keep in mind that input data should be similar to the transfer testing method (using 10% of data for example)