When using JAX don't import Numpy

Jammy2211 / PyAutoGalaxy

PyAutoGalaxy: Open-Source Multiwavelength Galaxy Structure & Morphology

https://pyautogalaxy.readthedocs.io/

MIT License

28 stars 14 forks source link

Closed CKrawczyk closed 3 months ago

CKrawczyk commented 4 months ago

Switches out numpy for jax.numpy. A very simple fix that makes radial profiles work as expected.