Softwares

I have either developed or contributed to following codes.

GWKokab

I am the developer and core maintainer of GWKokab, an open-source toolkit developed in collaboration with Mr. Muhammad Zeeshan and Dr. Richard O’Shaughnessy. This JAX-based framework is specifically engineered for high-performance gravitational-wave population inference, utilizing automatic differentiation and hardware acceleration to streamline Bayesian analysis for parametric models.

NumPyro

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.