pyro-ppl/numpyro

Deep universal probabilistic programming with Python and PyTorch

Bayesian Modeling and Probabilistic Programming in Python

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Probabilistic reasoning and statistical analysis in TensorFlow

Gaussian processes in TensorFlow

A highly efficient implementation of Gaussian Processes in PyTorch

Exploratory analysis of Bayesian models with Python

Bayesian inference with probabilistic programming.

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