Gwkokab

<div align="center"> <picture> <!-- Dark mode image --> <source srcset="https://raw.githubusercontent.com/gwkokab/gwkokab/main/docs/source/_static/noBgWhite.png" media="(prefers-color-scheme: dark)"> <!-- Light mode image --> <source srcset="https://raw.githubusercontent.com/gwkokab/gwkokab/main/docs/source/_static/noBgBlack.png" media="(prefers-color-scheme: light)"> <!-- Fallback image (if no preference detected) --> <img src="https://raw.githubusercontent.com/gwkokab/gwkokab/main/docs/source/_static/noBgColor.png" alt="logo"> </picture> </div> <h2 align="center"> A JAX-based gravitational-wave population inference toolkit for parametric models </h2>

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.

Citing GWKokab

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

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

@software{gwkokab2024github,
    author  = {{Qazalbash}, Meesum and {Zeeshan}, Muhammad and {O'Shaughnessy}, Richard},
    title   = {{GWKokab}: A JAX-based gravitational-wave population inference toolkit for parametric models},
    url     = {https://github.com/gwkokab/gwkokab},
    year    = {2024}
}