Vectorise Tiling functions

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

Closed CosmoMatt closed 1 year ago

CosmoMatt commented 1 year ago

Go through the functions within tiling.py and wherever possible move to a vectorised implementation.