Library implementation of https://arxiv.org/abs/1908.05968v5 - View it on GitHub