Check autodiff and batching support for healpix_fft_cuda
primitive and add if needed
#237
Labels
enhancement
New feature or request
I think the primitive added in #204 may not support automatic differentiation and batching transforms as we did not define Jacobian vector product and transpose operations (for autodiff) and a batcher (for
vmap
support). We should verify if this is the case and add implementations if necessary.The text was updated successfully, but these errors were encountered: