Tools for the inference of compartmental models

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Normalizing-flow enhanced sampling package for probabilistic inference in Jax

Bayesian Modeling and Probabilistic Programming in Python

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.

Statistical Bootstrap in Python

Inferring high-resolution human mixing patterns for disease modeling

Collection of bits and bops that I frequently use in my python projects

Calibrated inference of spiking from calcium ΔF/F data using deep networks

(Deprecated) Experimental PyMC interface for TensorFlow Probability. Official work on this project has been discontinued.

Bayesian python toolbox for inference and forecast of the spread of the Coronavirus

Toolbox for the Multistep Regression Estimator