@@ -35,7 +35,7 @@ def inverse(
3535 :math:`\alpha` with :math:`\phi`.
3636
3737 Should the user select method = "jax_ssht" they will be restricted to deployment on
38- CPU using our custom JAX frontend for the highly optimised SSHT C library [1]. In many
38+ CPU using our custom JAX frontend for the SSHT C library [1]. In many
3939 cases this approach may be desirable to mitigate e.g. memory i/o cost.
4040
4141 Args:
@@ -358,11 +358,11 @@ def inverse_jax_ssht(
358358) -> jnp .ndarray :
359359 r"""Compute the inverse Wigner transform (SSHT JAX).
360360
361- SSHT is a highly optimised C library which implements the spin-spherical harmonic
362- transform outlined in McEwen & Wiaux 2011 [1]. We make use of their python bindings
363- for which we provide custom JAX frontends, hence providing support for automatic
364- differentiation. Currently these transforms can only be deployed on CPU, which is a
365- limitation of the SSHT C package.
361+ SSHT is a C library which implements the spin-spherical harmonic transform outlined
362+ in McEwen & Wiaux 2011 [1]. We make use of their python bindings for which we
363+ provide custom JAX frontends, hence providing support for automatic differentiation.
364+ Currently these transforms can only be deployed on CPU, which is a limitation of the
365+ SSHT C package.
366366
367367 Args:
368368 flmn (jnp.ndarray): Wigner coefficients with shape :math:`[2N-1, L, 2L-1]`.
@@ -450,8 +450,8 @@ def forward(
450450 :math:`\alpha` with :math:`\phi`.
451451
452452 Should the user select method = "jax_ssht" they will be restricted to deployment on
453- CPU using our custom JAX frontend for the highly optimised SSHT C library [1]. In many
454- cases this approach may be desirable to mitigate e.g. memory i/o cost.
453+ CPU using our custom JAX frontend for the SSHT C library [1]. In many cases this
454+ approach may be desirable to mitigate e.g. memory i/o cost.
455455
456456 Args:
457457 f (np.ndarray): Signal on the on :math:`SO(3)` with shape
@@ -787,11 +787,11 @@ def forward_jax_ssht(
787787) -> jnp .ndarray :
788788 r"""Compute the forward Wigner transform (SSHT JAX).
789789
790- SSHT is a highly optimised C library which implements the spin-spherical harmonic
791- transform outlined in McEwen & Wiaux 2011 [1]. We make use of their python bindings
792- for which we provide custom JAX frontends, hence providing support for automatic
793- differentiation. Currently these transforms can only be deployed on CPU, which is a
794- limitation of the SSHT C package.
790+ SSHT is a C library which implements the spin-spherical harmonic transform outlined
791+ in McEwen & Wiaux 2011 [1]. We make use of their python bindings for which we
792+ provide custom JAX frontends, hence providing support for automatic differentiation.
793+ Currently these transforms can only be deployed on CPU, which is a limitation of the
794+ SSHT C package.
795795
796796 Args:
797797 f (jnp.ndarray): Signal on the on :math:`SO(3)` with shape
0 commit comments