Diffhalos: Differentiable Populations of Dark Matter Halos¶

Diffhalos is a generative model of cosmological volumes of dark matter halos, subhalos, and their merger trees. The Diffhalos library is written in JAX, and makes differentiable predictions for halo populations in a lightcone, or at a single redshift. The code is publicly available on GitHub.