JAXtronomy: A JAX port of lenstronomy
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2026
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866911739132510208 |
|---|---|
| author | Huang, Alan Birrer, Simon Gilman, Daniel Hogg, Natalie B. Shajib, Anowar J. Galan, Aymeric Zhang, Nan |
| author_facet | Huang, Alan Birrer, Simon Gilman, Daniel Hogg, Natalie B. Shajib, Anowar J. Galan, Aymeric Zhang, Nan |
| contents | Gravitational lensing is a phenomenon where light bends around massive objects, resulting in distorted images seen by an observer. Studying gravitationally lensed systems provides insights into cosmology and astrophysics, including constraints of the expansion rate of the Universe and the distribution of dark matter.
Thus, we introduce JAXtronomy, a re-implementation of the gravitational lensing software package lenstronomy (Birrer, 2021; Birrer & Amara, 2018) using JAX (Bradbury et al., 2018). JAX is a Python library that uses an accelerated linear algebra (XLA) compiler to improve the performance of computing software. Our core design principle of JAXtronomy is to maintain an identical API to that of lenstronomy.
The main JAX features utilized in JAXtronomy are just-in-time compilation, which can lead to significant reductions in execution time, and automatic differentiation, which allows for the implementation of gradient-based algorithms that were previously impossible. Additionally, JAX allows code to be run on GPUs or parallelized across CPU cores, further boosting the performance of JAXtronomy. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2606_01547 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | JAXtronomy: A JAX port of lenstronomy Huang, Alan Birrer, Simon Gilman, Daniel Hogg, Natalie B. Shajib, Anowar J. Galan, Aymeric Zhang, Nan Instrumentation and Methods for Astrophysics Cosmology and Nongalactic Astrophysics Gravitational lensing is a phenomenon where light bends around massive objects, resulting in distorted images seen by an observer. Studying gravitationally lensed systems provides insights into cosmology and astrophysics, including constraints of the expansion rate of the Universe and the distribution of dark matter. Thus, we introduce JAXtronomy, a re-implementation of the gravitational lensing software package lenstronomy (Birrer, 2021; Birrer & Amara, 2018) using JAX (Bradbury et al., 2018). JAX is a Python library that uses an accelerated linear algebra (XLA) compiler to improve the performance of computing software. Our core design principle of JAXtronomy is to maintain an identical API to that of lenstronomy. The main JAX features utilized in JAXtronomy are just-in-time compilation, which can lead to significant reductions in execution time, and automatic differentiation, which allows for the implementation of gradient-based algorithms that were previously impossible. Additionally, JAX allows code to be run on GPUs or parallelized across CPU cores, further boosting the performance of JAXtronomy. |
| title | JAXtronomy: A JAX port of lenstronomy |
| topic | Instrumentation and Methods for Astrophysics Cosmology and Nongalactic Astrophysics |
| url | https://arxiv.org/abs/2606.01547 |