gwkokab 0.2.0

A JAX-based gravitational-wave population inference toolkit for parametric models

A JAX-based gravitational-wave population inference toolkit for parametric models

Installation | Documentation | Examples/Tutorials | FAQs | Citing GWKokab

GWKokab is a JAX-based gravitational-wave population inference toolkit. It is designed to be a high-performance, flexible, easy-to-use library for sampling from a wide range of gravitational-wave population models. It is built on top of JAX, a high-performance numerical computing library, and is designed to be easily integrated into existing JAX workflows.

If you would like to contribute, please see the contributing guidelines.

[!WARNING] Our documentation is generally maintained, but tutorials are currently outdated. While the main documentation is largely current, some inconsistencies may exist. We greatly value community feedback – if you encounter any discrepancies, please submit an issue report to help us improve the documentation quality for all users.

Citing GWKokab

If you use GWKokab in your research, please cite the following paper:

@article{arxiv:2509.13638,
    author  = {{Qazalbash}, Meesum and {Zeeshan}, Muhammad and {O'Shaughnessy}, Richard},
    title   = {GWKokab: An Implementation to Identify the Properties of Multiple Population of Gravitational Wave Sources},
    journal = {arXiv preprint arXiv:2509.13638},
    year    = {2025},
    url     = {https://arxiv.org/pdf/2509.13638v1}
}