@Bartosz_Kurkowski I’m not sure if your network architecture matches the one stated in the paper (as details are very scarce in it).
I would also like to implement this network in Pytorch. I have looked at Gluon-ts implementation (as it is the one written by the paper’s authors), but I am very unfamiliar with mxnet/gluon…
Have you find another reference implementation before writinf your own?