← Back to Cosmology
astro-ph.CO

Can faster software unlock new secrets from gravitational lensing?

Alan Huang, Simon Birrer, Daniel Gilman, Natalie B. Hogg, Anowar J. Shajib, Aymeric Galan, Nan Zhang

June 1, 2026

Gravitational lensing—light bending around massive objects—reveals dark matter and cosmic expansion rates. A new implementation of the lenstronomy software in JAX, a machine learning library with GPU acceleration and automatic differentiation, maintains the same interface while dramatically speeding computation. This unlocks gradient-based inference methods that were computationally intractable before.
Published as JAXtronomy: A JAX port of lenstronomy arXiv:2606.01547
Read the original paper →