## 🚀 Feature Request To make sure that our implementation is similar to the base paper's one, we should implement their benchmarks and compare performance. Paper link: https://arxiv.org/pdf/1812.02356.pdf