-
Notifications
You must be signed in to change notification settings - Fork 8
demo1-jax.py #92
Copy link
Copy link
Open
Milestone
Description
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)
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels