ENH enable fixed size photon arrays

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 enable fixed size photon arrays #83

Closed beckermr closed 9 months ago

beckermr commented 9 months ago

This PR adds APIs to enable photon arrays that have a fixed underlying size but represent a variable number of photons. This will hopefully help in apply JIT to sections of code where Poisson sampling is used.