Reimplementation of KSR in pytorch version. The original is written in theano (https://github.com/BetsyHJ/KSR).