GitLab

* fixing banding issues with corrected computation of kernel size
* fixing Parameters
* changed normalization of isotropic kernels

---------
Co-authored-by: Thorsten Kurth <tkurth@nvidia.com>

* working distributed fwd SHT with flexible sharding and without padding

* fixing distributed sht with new logic

* fixed distributed SHT and ISHT with flexible padding

* working unittest for distributed fwd SHT

* distributed tests converted to unit tests

* bumping version number up to 0.6.4

* fixing small splitting bug in th distributed

* updated changeloc, removed tests folder

* Adding prototype implementation of disco convolutions

* adding s2convolutions.py

* Somewhat functional first implementation in Triton

* Somewhat functional first implementation in Triton

* Made batched execution work on GPU and refactored precomputation of kernel values and supports.

* Adding the Lobatto grid to the grid selection method in quadrature

* fixing bug in triton kernel when iterating over non-zeros

* Updating with working custom triton implementation of DISCO convolution

* Merged both forward and backward DISCO contraction into a single kernel

* Some cleanup and minor bugfix

* bugfixes to the reference implementation

* adding weights to s2conv

* removing unnecessary imports

* suggestion for torch harmonics, math should be checked thoroughly

* Intermediate working reference implementation for the transpose DISCO convolution. Fixed normalization of kernels

* adjusting cutoff frequency in disco convolution

* fixing transpose DISCO contraction

* moving triton to install_requires

* enabling triton by default

---------
Co-authored-by: Thorsten Kurth <tkurth@nvidia.com>

* Adding prototype implementation of disco convolutions

* adding s2convolutions.py

* Somewhat functional first implementation in Triton

* Somewhat functional first implementation in Triton

* Made batched execution work on GPU and refactored precomputation of kernel values and supports.

* Adding the Lobatto grid to the grid selection method in quadrature

* fixing bug in triton kernel when iterating over non-zeros

* Updating with working custom triton implementation of DISCO convolution

* Merged both forward and backward DISCO contraction into a single kernel

* Some cleanup and minor bugfix

* bugfixes to the reference implementation

* adding weights to s2conv

* removing unnecessary imports

* suggestion for torch harmonics, math should be checked thoroughly

* Intermediate working reference implementation for the transpose DISCO convolution. Fixed normalization of kernels

* adjusting cutoff frequency in disco convolution

* fixing transpose DISCO contraction

* moving triton to install_requires

---------
Co-authored-by: Thorsten Kurth <tkurth@nvidia.com>

* Update README.md

adding autocast remark to readme

* Update README.md

small typo fix

* Adding notebooks for gradient analysis

* Updated the Readme

* Refactored training script for sfno

* removing concatanated skip connections

* Added some logic for initializing weights adaptively

* Cleanup and fixing initialization in SFNO. Removing non-linear variant of SFNO

* Updated initialization scheme for SFNO

* Some more cleanup in SFNO

* updated the SFNO architecture

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

* Adding notebooks for gradient analysis

* Updated the Readme

Fixing issues in README.md

* Adding plotting routines and reworked Readme

* Added illustration of spherical harmonics to Readme

* Updated Changelog

* improved readability of legendre.py

* bugfix in computation of dlegpoly

* Moving conversion from numpy to torch from the legendre module to the sht module

* renaming to vdm in tests.py

* fixing contig in distributed sht

* make it normally contiguous before all to all

* commented the code to be a WAR

* reworked SFNO example

* updated changelog

* Added gradient check to test suite
* reduced size of the unit test
* switched to parametrized for unittests

fixed error in getting started notebook

Adding missing dependency to functoools.partial in sfno.py

Fix __init__.py file to import from local "examples" dir.

Fixing precision mismatch error in weight contractions

Without this fix, importing torch_harmonics had run into a module import error
when trying to execute:
```import examples```

Update __init__.py