Bayesian Modeling and Probabilistic Programming in Python
Deep universal probabilistic programming with Python and PyTorch
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
A neurosymbolic perspective on LLMs
High-quality implementations of standard and SOTA methods on a variety of tasks.
BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
Gaussian processes in JAX and Equinox.