A GPU-optimized implementation of Tucker-factorized Fourier Neural Operator from Multi-Grid Tensorized Fourier Neural Operator for High-Resolution PDEs.
A GPU-optimized implementation of Tucker-factorized Fourier Neural Operator from Multi-Grid Tensorized Fourier Neural Operator for High-Resolution PDEs.