ENH k-space wrapping for Hermitian images

GalSim-developers / JAX-GalSim

JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.

Other

25 stars 3 forks source link

ENH k-space wrapping for Hermitian images #65

Closed beckermr closed 10 months ago

beckermr commented 11 months ago

This PR has a very naive, but working implementation of k-space wrapping for Hermitian images. The code expands the image from its compressed format to the fully explicit Hermitian format, wraps then using simple code, and then cuts away the parts not needed.

I tried a more memory efficient algorithm that didn't involve the expansion but could not get it working. I've left that code commented out in case someone else can find the bug.

todo:

[x] add tests of reverse-mode autodiff

beckermr commented 10 months ago

I am merging this code with the fixes to address the PR comments above into PR #60 since it is needed for tests there.