For following a line of research that augments attention with a differentiable stack, beginning with DuSell et al. at ETH Zurich - View it on GitHub