TractorJAX: GPU-Accelerated SPHEREx Forced Photometry and the Trade-offs of Regularization and External Priors

Abstract

We present TractorJAX, an open-source GPU reimplementation of the catalog-forced mode of The Tractor. We use it to test whether regularization or external flux priors enable reliable fitting of reference catalogs that extend below the depth directly constrained by undersampled SPHEREx images. For full-depth regularized fits, TractorJAX fits $100\times100$-pixel cutouts at 11.9 ms each on a single NVIDIA H100 GPU. In simulated images of the Abell 2537 galaxy-cluster field, extending the fitted ca…