Iterative refinement support for JAX and NumPy forward (spherical) transform implementations

[matt-graham]

Should resolve #139

Adds support for using iter keyword argument to s2fft.forward when using method = "numpy" and method = "jax", using an iterative refinement approach equivalent to that implemented in healpy to improve accuracy of forward transform (in terms of acting as the inverse of the inverse transform).

Currently this is only added for spherical transform, and is available for all sampling types, even though in practice probably only useful for HEALPix (though iter=1 does seem to give a small gain in accuracy for other sampling types).

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 96.11%. Comparing base (909e6f1) to head (bb04f00).

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #241      +/-   ##
==========================================
+ Coverage   96.07%   96.11%   +0.04%     
==========================================
  Files          31       31              
  Lines        3567     3581      +14     
==========================================
+ Hits         3427     3442      +15     
+ Misses        140      139       -1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[CosmoMatt]

@matt-graham looks great, one thing that would also be good to add is the same iterations but for the precompute versions of the transforms.

[jasonmcewen]

Good point @CosmoMatt (re precompute). We could either do that in the current PR if quick or else do it in a next one.