mega-pytorch (pypi): Mega - Pytorch
Keywords: artificial intelligence|deep learning|attention mechanism|exponential moving average|long range arena|artificial-intelligence|attention-mechanisms|deep-learning|exponential-moving-average|long-range-arena