GIGA-Lens 2.0: Strong-Lens Modeling on Multiple GPU Nodes

arXiv:2606.30633 · astro-ph.CO, astro-ph.GA · Submitted 2026-06-29 · Read on arXiv

Xiaosheng Huang, Linus Upson, Nicolas Ratier-Werbin, Harry Lu, Sean Xu, Elden Yap, Evan Odell, Ansel Parke, Harsh Ambardekar, Saul Baltasar, Nestor Demeure, Bradley Richardson, Andi Gu, Yuan-Ming Hsu, Junyi Liu

astro-ph.CO, astro-ph.GA

Submitted: 2026-06-29

Comments: 21 pages, 8 figures, and 3 tables

Code: https://github.com/oogle/jax

License: http://creativecommons.org/licenses/by/4.0/

The gist: We present GIGA-Lens 2.0: a major upgrade to the GPU-accelerated Bayesian framework for modeling strong lensing systems that allows it to be run across multiple GPU nodes.

Terminology

• Adam — 11× in this paper · explained in 2 episodes
• Adaptation — 5× in this paper · explained in 1 episode
• Department of Energy — 4× in this paper
• Bayesian Framework — 3× in this paper · explained in 3 episodes
• Hamiltonian Monte Carlo — 3× in this paper
• National Energy Research Scientific Computing Center — 3× in this paper
• Bayesian Inference — 2× in this paper · explained in 5 episodes
• Gravitational Lensing — 2× in this paper · explained in 2 episodes
• Accuracy — 1× in this paper · explained in 1 episode
• Algorithmic Improvements — 1× in this paper · explained in 1 episode
• Fidelity — 1× in this paper · explained in 2 episodes
• Forward modeling — 1× in this paper · explained in 2 episodes
• Objective Function — 1× in this paper · explained in 2 episodes
• Parallelization for SVI Stochastic Variational Inference — 1× in this paper
• Robustness — 1× in this paper · explained in 5 episodes

Abstract

Sources

• A Novel Lensed Point Source Modeling Pipeline using GIGA-Lens with Application to SN Zwicky and SN iPTF16geu
• TDCOSMO IV: Hierarchical time-delay cosmography -- joint inference of the Hubble constant and galaxy density profiles
• TensorFlow Distributions
• Demystifying NCCL: An In-depth Analysis of GPU Communication Protocols and Algorithms
• DESI Strong Lens Foundry V: A Sample of HST-Observed Strong Lenses Modeled with GIGA-Lens
• Nested $\hat R$: Assessing the convergence of Markov chain Monte Carlo when running many short chains
• The Carousel Lens I: A Spectroscopic Survey of the Carousel Lens Field
• The Carousel Lens II: Cosmological Constraints with GIGA-Lens
• AdaBelief Optimizer: Adapting Stepsizes by the Belief in Observed Gradients

Related papers

• Angular clustering and bias of photometric quasars in the Kilo-Degree Survey Data Release 4
• A Novel kinetic Sunyaev-Zel'dovich Estimator for Electron-Electron Correlations
• Magnetic fields at the dawn of structure formation I. The CARLA J1510+5958 proto-cluster
• Dark Energy Survey Year 6 Results: Weak Lensing and Galaxy Clustering Cosmological Analysis Framework
• Exploring the Impact of Systematic Bias in Type Ia Supernova Cosmology Across Diverse Dark Energy Parametrizations
• Non-Gaussian Galaxy Stochasticity and the Noise-Field Formulation