halox: Dark matter halo properties and large-scale structure calculations using JAX
Abstract
Dark matter halos are fundamental structures in cosmology, forming the gravitational potential wells hosting galaxies and clusters of galaxies. Their properties and statistical distribution (including the halo mass function) are invaluable tools to infer the fundamental properties of the Universe. The \texttt{halox} package is a JAX-powered Python library enabling differentiable and accelerated computations of key properties of dark matter halos, and of the halo mass function. The automatic differentiation capabilities of \texttt{halox} enable its usage in gradient-based workflows, e.g. in efficient Hamiltonian Monte Carlo sampling or machine learning applications.
Keywords
Cite
@article{arxiv.2509.22478,
title = {halox: Dark matter halo properties and large-scale structure calculations using JAX},
author = {Florian Kéruzoré},
journal= {arXiv preprint arXiv:2509.22478},
year = {2025}
}
Comments
6 pages, submitted to the Journal of Open Source Software. halox is available at https://github.com/fkeruzore/halox