Open
Description
In
s2fft/s2fft/recursions/trapani.py
Lines 776 to 792 in ea11832
the entire body of the function is commented out. I suspect this may be something to do with the assert
statements not working correctly within a JAX JIT compiled function due to needing to know the concrete shapes of arrays given the commenting out was done in the changes in 3b514f1 adding JAX implementation by @jasonmcewen. We should either remove the function if no longer required or update to make it work if the issue is with JAX JIT transform.
Activity