Lya2pcf: an efficient pipeline to estimate two- and three-point correlation functions of the Lyman-$α$ forest

Abstract

Studying the matter distribution in the universe through the Lyman-$α$ forest allows us to constrain small-scale physics in the high-redshift regime. Spectroscopic quasar surveys are generating increasingly large datasets that require efficient algorithms to compute correlation functions. Moreover, cosmological analyses based on Lyman-$α$ forests can significantly benefit from incorporating higher-order statistics alongside traditional two-point correlations. In this work, we present Lya2pcf, a pipeline designed to compute three-dimensional two-point and three-point correlation functions using Lyman-$α$ forest data. The code implements standard algorithms widely used in current spectroscopic surveys for computing the two-point correlation function with its distortion matrix, covariance matrices; and it naturally extends the two-point estimator to three-point correlations. Thanks to GPU optimization, Lya2pcf achieves a substantial reduction in computational time for both the two-point correlation function and its distortion matrix when compared to the widely used PICCA code. We apply Lya2pcf to data from the Sloan Digital Sky Survey (SDSS) sixteenth data release (DR16) and a Dark Energy Spectroscopic Instrument Year-5 (DESI Y5) mock dataset, demonstrating overall performance gains over PICCA, especially on GPUs. We show the first measurement of the anisotropic three-point correlation function on a large spectroscopic sample for all possible triangles with scales up to 80 Mpc/h. The estimator's fast computation and the resulting signal-to-noise ratio -- above one for many triangle configurations -- demonstrate the viability of incorporating three-point statistics into future cosmological inference analyses, particularly with the larger datasets expected from Stage IV spectroscopic surveys.