adirondax

A differentiable astrophysics simulator in JAX

Author: Philip Mocz (@pmocz)

⚠️ Adirondax is currently being built and is not yet ready for use. Check back later ⚠️

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 Adirondax

Install with

pip install adirondax

See the docs for more info on how to build from source.

Examples

Check out the examples/ directory for demonstrations of using Adirondax.

Links

• 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).

Cite this repository

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}
}