Adaptive FNO transformer - official Pytorch implementation - View it on GitHub