This is a selection of open-source and research-related software I’ve developed.
You can also find my GitHub for details.

Highlighted Open-Source Libraries

Bayex: Bayesian Optimization in JAX
Open GitHub

Simple yet powerful Bayesian Optimization library using JAX and XLA compilation. Ideal for black-box optimization in research workflows.
PCAx: Differentiable PCA in JAX
Open GitHub

A lightweight, differentiable PCA implementation in JAX, designed for seamless GPU acceleration. Enables backpropagation through PCA layers with minimal overhead.
t-SNEx: Minimal t-SNE in JAX
Open GitHub

High-performance, minimal t-SNE implementation for visualizing high-dimensional data. Built with scalability and GPU efficiency in mind.
BoundVor: Bounded Voronoi Tessellation
Open GitHub

Wrapper around scipy.spatial.Voronoi to clip diagrams to arbitrary polygons. Useful for modeling bounded spatial domains in physical systems.

Highlighted Research Code

De(ep)tangle: High-Density Worm Tracking
Open GitHub

State-of-the-art tracking system for overlapping slender bodies in biological images. Published in Nature Communications Biology (2023). → Read the paper
ChemoXRL: Reinforcement Learning for Chemotaxis
Open GitHub

Deep RL (PPO) implementation for high-noise environments in cellular navigation. Used in PNAS Nexus (2024) and related works. → Read the paper
Recloc: Receptor Clustering and Sensing Accuracy
Open GitHub

Simulation framework and analysis toolkit for studying spatial receptor clustering and its effect on sensing performance. Used in the PRL 2025 paper: Local clustering but global spreading of receptors for optimal sensing. → Read the paper
GradNodes: Adaptive Node in Transport Networks
Open GitHub

Differentiable framework to explore optimal node positioning in spatial networks under flow constraints. Part of the work presented in the 2024 preprint on adaptive node placement. → Read the paper