Furax: A Modular JAX Framework for Linear Operators in Astrophysical and Cosmological Data Analysis

TL;DR

Furax is an open-source JAX-based framework that offers modular linear operators and solvers tailored for astrophysical and cosmological data analysis, facilitating modeling and inverse problem solving.

Contribution

It introduces a modular, extensible Python framework built on JAX for modeling data systems and solving inverse problems in astrophysics and cosmology.

Findings

Provides domain-specific tools for CMB data analysis

Enables efficient modeling of data acquisition systems

Supports complex inverse problem solutions

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 analysis$-$including map-making, instrument modeling, and astrophysical component separation$-$with a modular architecture designed to extend to other fields.