Showcase how to use jax functionality in jax galsim, specifically: - [ ] jitting - [ ] vmapping - [ ] gradients Potential cool extensions (potentially other demos) - [ ] Derivatives with respect to shear `d Image / d Shear` - [ ] running gradient based samplers with jax-galsim (eg blackjax, numpyro) - [ ] jax metacal (see unit test) - [ ] full stamp image drawing with jax galsim (see unit test)
Showcase how to use jax functionality in jax galsim, specifically:
Potential cool extensions (potentially other demos)
d Image / d Shear