Skip to content

demo1-jax.py #92

@ismael-mendoza

Description

@ismael-mendoza

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)

Metadata

Metadata

Assignees

No one assigned

    Labels

    docsImprovements or additions to documentation

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions