GitLab

* removing duplicate code from distributed convoloution

* replacing from_numpy with as_tensor

* make preprocess_psi_tensor GPU ready.

* setting imaginary parts of DCT and nyquist frequency to zero in IRSHT variants

* Improved computation of Morlet filter basis and switched to a Hann window.

* addresses #064 and some cleanup

adding local SNO architecture and updating trraining script for SWE prediction with powerspectrum computation

* changing distributeds SHT to use dim=-3 as the channel dimension for distributed transpose

* Fixed formatting in tests

* Adding bash script for running test suite

* Replaced asserts regarding number of dimensions in tensor with checks

* AMP hotfix

* Bumping up version to 0.7.1

Major Cleanup of SFNO. Retiring non-linear architecture and fixing initialization. Adding scripts for training and validation.

* reworked SFNO example

* updated changelog