Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
MCMC samplers based on simulating Hamiltonian dynamics on a manifold.
Library to run VI algorithms on Stan models.
Consistent Optimization of Label-wise Utilities in Multi-label classificatioN
