It looks like the bulk of the code for training the EMI elimination model is is pretty straightforward PyTorch.