Furax: A Modular JAX Framework for Linear Operators in Astrophysical and Cosmological Data Analysis
Instrumentation and Methods for Astrophysics
2026-03-23 v1 Cosmology and Nongalactic Astrophysics
Abstract
The Framework for Unified and Robust data Analysis with JAX (Furax) is an open-source Python framework for modeling data acquisition systems and solving inverse problems in astrophysics and cosmology. Built on JAX, Furax provides composable building blocks in the form of general-purpose and domain-specific linear operators, along with preconditioners and solvers for their numerical inversion. Domain-specific tools are provided for astrophysical and cosmic microwave background (CMB) data analysisincluding map-making, instrument modeling, and astrophysical component separationwith a modular architecture designed to extend to other fields.
Keywords
Cite
@article{arxiv.2603.19600,
title = {Furax: A Modular JAX Framework for Linear Operators in Astrophysical and Cosmological Data Analysis},
author = {Pierre Chanial and Simon Biquard and Wassim Kabalan and Wuhyun Sohn and Artem Basyrov and Benjamin Beringue and Alexandre Boucaud and Andréa Landais and Magdy Morshed and Radek Stompor and Ema Tsang King Sang and Amalia Villarrubia-Aguilar and Josquin Errard},
journal= {arXiv preprint arXiv:2603.19600},
year = {2026}
}
Comments
Submitted to the Journal of Open Source Software