a pytorch implementation of https://arxiv.org/abs/1704.03477 - View it on GitHub