GalSim-developers / JAX-GalSim

JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.
Other
25 stars 3 forks source link

demo1-jax.py #92

Open ismael-mendoza opened 3 months ago

ismael-mendoza commented 3 months ago

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: