JAX-GalSim
JAX-GalSim copied to clipboard
JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.
It appears something changed in how jax is seeding their RNGs and now tests are failing.
I've heard reports that the spergel profiles are much slower than galsim. To Do: - [x] add benchmark comparison to galsim - [ ] profile to find optimizations - [x]...
Jax-Galsim import fails with jax 0.7 due to package compatibility issues. Here is the stack trace of the error: ```deep_field_metadetect/jaxify/observation.py:4: in import jax_galsim ../../../micromamba/envs/dfmdet/lib/python3.11/site-packages/jax_galsim/__init__.py:55: in from .moffat import Moffat ../../../micromamba/envs/dfmdet/lib/python3.11/site-packages/jax_galsim/moffat.py:7:...