Meta: Ability to use (in FTorch) other neural network architectures e.g. CNN, FNOs

[omarjamil]

No description provided.

[jatkinson1000]

Input from DW as to what these would be.

@yqsun91 comments: ANN, CNN, and FNO

@jatkinson1000 comments main thing to check is whether we can run these PyTorch features through TorchScript. To check FNO we would need to check this. @HamidPahlavan has an example of a FNO - if this can be shared then we can explore how easy it is to save this as TorchScript: https://github.com/HamidPahlavan/Nonlocality/blob/main/FNO.ipynb

[jatkinson1000]

Suggestion is that we fork Hamid's repo above and create a Torchscript versioin of the CNN and FNO, write a Fortran implementation using the same NetCDF data, and then compare the results.

These are broken into #23 and #24