astro-informatics / s2wav

Differentiable and accelerated wavelet transform on the sphere with JAX
https://astro-informatics.github.io/s2wav/
MIT License
14 stars 0 forks source link

Core Dimension Helper Functions #25

Closed JessWhitney closed 1 year ago

JessWhitney commented 1 year ago

Code for issue #8 . Includes three sections of code:

I've also tried to include a brief description for each function, but for some of the functions towards the end I wasn't 100% sure what they were doing, so I expect I will have to alter them after review.

Note: I accidentally saved both this and the parameters class function to this branch. I have since deleted that file (hence, I think, why it says 3 commits rather than 1) but hopefully this PR now only includes the relevant file.