JAX-GalSim

JAX-GalSim copied to clipboard

validating class instantiation and jitting

5

In many classes in Galsim there is code that validates the instantiation of a class object. This is difficult to maintain if we want make the class jittable and vmappable....

ismael-mendoza

Setup documentation system to keep track of implemented API

1

This is an issue for sphinx-savvy people who love to setup documentation for their Python projects :-) We need to add automated doc building, which we could host on read...

EiffL

[pre-commit.ci] pre-commit autoupdate

1

updates: - [github.com/psf/black: 24.4.2 → 24.8.0](https://github.com/psf/black/compare/24.4.2...24.8.0) - [github.com/pycqa/flake8: 7.0.0 → 7.1.1](https://github.com/pycqa/flake8/compare/7.0.0...7.1.1)

pre-commit-ci[bot]

use codspeed to benchmark performance and look for regressions

2

We can use this free service to write benchmark tests for CPUs: https://docs.codspeed.io/benchmarks/python We won't be able to test on GPUs, but it will be a good start.

beckermr

Tasks related to releasing a first version of the code along with a companion paper - [x] #88 - [ ] #90 - [ ] #91 - [ ] #92...

ismael-mendoza

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...

ismael-mendoza

Copy over and ensure that we can run demo2.py from Galsim: https://github.com/GalSim-developers/GalSim/blob/releases/2.5/examples/demo2.py We are missing `galsim.hsm.EstimateShear(image, image_epsf)`, but we could use the GalSim one here

ismael-mendoza

Copy over and ensure that we can run demo1.py from Galsim: https://github.com/GalSim-developers/GalSim/blob/releases/2.5/examples/demo1.py We are missing `image.FindAdaptiveMom()` but we could use the GalSim one (or implement if not too complicated).

ismael-mendoza

ensure that it accurately mentions currently code capabilities, gotchas, and links to documentation (to be done after #88 and #17 )

ismael-mendoza

- [ ] #71 - [ ] commented out code - [ ] places where black formatting makes code particularly difficult to read (refactor into multiple lines) e.g. [here](https://github.com/GalSim-developers/JAX-GalSim/blob/4b12d6b3520938cd823ae3978c400bb9a2b454d3/jax_galsim/bessel.py#L13) -...

ismael-mendoza