Risbo JAX

s2fft.recursions.risbo_jax.compute_full(dl: Array, beta: float, L: int, el: int) → Array

Compute Wigner-d at argument \(\beta\) for full plane using Risbo recursion (JAX implementation).

The Wigner-d plane is computed by recursion over \(\ell\) (el). Thus, for \(\ell > 0\) the plane must be computed already for \(\ell - 1\). At present, for \(\ell = 0\) the recusion is initialised.

Parameters:

• dl (jnp.ndarray) – Wigner-d plane for \(\ell - 1\) at \(\beta\).

• beta (float) – Argument \(\beta\) at which to compute Wigner-d plane.

• L (int) – Harmonic band-limit.

• el (int) – Spherical harmonic degree \(\ell\).

Returns:

Plane of Wigner-d for el and beta, with full plane computed.

Return type:

jnp.ndarray