Softwares

gwkokab/gwkokab

I am the developer and core maintiner of this project. It was build in collaboration with Mr Muhammad Zeeshan and Dr Richard O'Shaughnessy. It is an open-source JAX-based gravitational-wave population inference toolkit for parametric models.

pyro-ppl/numpryo

I have been a top contributor to NumPyro since February 2024. My contributions include the implementation of new probability distributions, resolving issues related to automatic differentiation, and addressing bugs originating from upstream dependencies such as JAX.

In February 2026, I was invited to become an official collaborator on the NumPyro repository (LinkedIn post). In this role, I contribute to maintaining the project’s alignment with modern infrastructure and best practices, while ensuring its continued stability and relevance within the probabilistic programming ecosystem.

A summary of my contributions can be found here.