GitLab

fixing bug in quadrature weights for full attention. Adding better unit tests for attention. Cleanup in the cuda code.

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

This reverts commit 82881276.

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

* small fix

* making einsum result contiguous

* adding zero frequency to distributed sht

* small hotfix for lobatto grid precomputation routine

* adding lobatto grid to tests

* adding caching

* replacing many numpy calls with torch calls

* bumping up version number to 0.7.6

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

* addresses #064 and some cleanup

Without putting `signal` on `device`, we get `RuntimeError: Expected all tensors to be on the same device` when `sht` is called.

switched psi tensor computation to double precision and implemented a fudge factor for theta_cutoff to avoid aliasing issues with the grid width

updated resampling module to handle corner cases where the input signal needs to be extended towards the pole

Adding FilterBasis datastructure to contain information about the type of basis used in disco convolution

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