adirondax

adirondax#

Differentiable Astrophysics
Adirondax logo

Adirondax is a scientific research software for conducting astrophysical and cosmological simulations and solving inverse problems, implemented in JAX and utilizing automatic differentiation and multi-GPU acceleration.

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

Differentiable Physics

Solvers in Adirondax are fully differentiable, enabling gradient-based inference, parameter optimization, and coupling to ML models directly through the simulation.

Scalable Performance

Built with JAX, Adirondax scales on multiple GPUs/TPUs, allowing large-scale simulations to run efficiently on modern accelerators.

Modular & Composable

Adirondax physics modules and processing tools are composable, making it easy to extend or embed into larger workflows.

Installation
Installation
Quickstart
Quickstart
Examples
Examples

Getting Started

Tutorials & Examples