Cool! I find it has been implemented in the tslib (https://github.com/thuml/Time-Series-Library), the results seem promising when I reproduce the experiments.