[In-Progress] Recurrent Weighted Average Implementation in Pytorch - View it on GitHub