GitLab

Please review @thomwolf but i think this is equivqlent (and it mimics the loss computation of the original loss)