pytorch

neuraloperator Tool

The open-source PyTorch library for Fourier neural operators and related architectures -- the toolkit behind FourCastNet and the plasma and lithography surrogates. If you want to try learning a solution operator for a PDE instead of solving it step by step, this is the reference implementation.