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.
- Flow-based variational inference for gravitational-wave population analysis.
See the examples.
- gwax is being actively developed and the interface is likely to undergo breaking changes.
- Currently the only thing implemented is variation inference of gravitational-wave populations.
- Variational population inference only interfaces with gwpopulation.
- Only univariate uniform priors are supported.