A differentiable astrophysics simulator in JAX
Author: Philip Mocz (@pmocz)
Adirondax is a high-performance scientific research software for conducting astrophysical and cosmological simulations. Being differentiable, Adirondax can seamlessly integrate with pipelines for inverse-problems, inference, optimization, and coupling to ML models. Adirondax is scalable on multiple GPUs.
Adirondax has a simpler companion project dedicated to Fuzzy Dark Matter simulations: Jaxion
Install with
pip install adirondaxSee the docs for more info on how to build from source.
Check out the examples/ directory for demonstrations of using Adirondax.
- Code repository on GitHub (this page).
- Documentation for up-to-date information about installing and using Adirondax.
- asv benchmarks to view code timing and memory benchmarks (for Developers).
If you use this software, please cite it as below.
@software{Mocz_Adirondax_2025,
author = {Mocz, Philip},
doi = {10.5281/zenodo.17584122},
month = nov,
title = {{Adirondax}},
url = {https://github.com/AdirondaxProject/adirondax},
version = {0.0.4},
year = {2025}
}