astro-informatics / s2wav

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

Vectorised Analysis Transform #39

Closed CosmoMatt closed 1 year ago

CosmoMatt commented 1 year ago

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

CosmoMatt commented 1 year ago

@JessWhitney this might be a good next issue for you to pick up!

JessWhitney commented 1 year ago

@JessWhitney this might be a good next issue for you to pick up!

Yeah I'll take a look at it this week, will be a good opportunity for me to understand vectorisation better.