Like this or something different?

https://github.com/gzerveas/mvts_transformer