Future Work

Add distributed support for multiple variations of the FNO such as

  • Mixed Precision FNO (Colin White et al., 2023)
  • GINO (Zongyi Li et al., 2023)
  • MGTFNO (Jean Kossaifi et al., 2023)
  • GANO (Md Ashiqur Rahman et al., 2022)
  • AFNO (John Guibas et al., 2021)
  • UFNO (Gege Wen et al., 2021)