Implementation of the Remixer Block from the Remixer paper, in Pytorch - View it on GitHub