Thanks for your impression, your idea is likely to be implemented! congrats! well, I won't put it in my github, but you must have a look at this repo (github.com/thuml/Autoformer) for your work. It is complete, solid and, perfect.