(partial) replication of results from https://arxiv.org/abs/1912.07768 - View it on GitHub