Fully featured implementation of Routing Transformer - View it on GitHub