I’m not familiar with this paper, but the authors mention they are using a Neural Network so based on this statement it should be possible to reimplement it using PyTorch. Do you have a specific dataset you are working with?