2025ApJ...994..144M -
Astrophys. J., 994, 144 (2025/December-1)
microJAX: A Differentiable Framework for Microlensing Modeling with GPU-accelerated Image-centered Ray Shooting.
MIYAZAKI S. and KAWAHARA H.
Abstract (from CDS):
We introduce microJAX, the first fully differentiable implementation of the image-centered ray shooting algorithm for gravitational microlensing. Built on JAX and its XLA just-in-time compiler, microJAX exploits GPU parallelism while providing exact gradients through automatic differentiation. The current release supports binary- and triple-lens geometries, including limb-darkened extended-source effects, and delivers magnifications that remain differentiable for all model parameters. Benchmarks show that microJAX matches the accuracy of established packages and attains up to a factor of ∼5-6 speedup in the small-source, limb-darkened regime on an NVIDIA A100 GPU. Since the model is fully differentiable, it integrates seamlessly with probabilistic programming frameworks, enabling scalable Hamiltonian Monte Carlo and variational inference workflows. Although the present work focuses on standard microlensing magnification models, the modular architecture is designed to support upcoming implementations of microlensing higher-order effects, while remaining compatible with external likelihood frameworks that incorporate advanced noise models. microJAX thus provides a robust foundation for precise and large-scale surveys anticipated in the coming decade, including the Nancy Grace Roman Space Telescope, where scalable, physically self-consistent inference will be essential for maximizing scientific return.
Abstract Copyright:
© 2025. The Author(s). Published by the American Astronomical Society.
Journal keyword(s):
Gravitational microlensing - Binary lens microlensing - Triple lens microlensing
Simbad objects:
1
Full paper
View the references in ADS
To bookmark this query, right click on this link: simbad:2025ApJ...994..144M and select 'bookmark this link' or equivalent in the popup menu