That's a good repo. It uses mxnet right? The aim of the repo in the topic was mainly to provide a clean implementation that could slot easily into an existing pytorch workflow.