Gravitational-wave astronomy in JAX.
- Install JAX following their instructions for your platform and hardware.
pip install gwax.
To install the latest development: pip install git+https://github.com/mdmould/gwax.
- Helper functions to load LVK data downloaded using
lvk-data. - Likelihood for gravitational-wave population analysis, including a new implementation that avoids single-event downsampling.
- Some standard parametric population models.
- Nonparametric population models.
- Flow-based variational inference, e.g., for population analyses.
See the examples.
Note that I made this repo primarily to contain code I reuse many times, not as a comprehensive and documented package. Please get in touch if you'd like help using it 👍