GitLab

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