JAX-GalSim icon indicating copy to clipboard operation
JAX-GalSim copied to clipboard

Published 20 hours ago • verified publisher icon GalSim-developers

demo1-jax.py

Open ismael-mendoza opened this issue 1 year ago • 0 comments

We probably don't have to do this, but we could consider adding somethign showcasing the jax-specific things that are enabled by using jax-galsim. Some ideas include:

  • [ ] Vectorization and jitting of image rendering functions
  • [ ] Derivatives with respect to galaxy parameters or shear d Image / d Shear
  • [ ] numpyro and HMC

ismael-mendoza avatar May 30 '24 20:05 ismael-mendoza