inference-algorithms
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Manifold Markov chain Monte Carlo methods in Python
Library to run VI algorithms on Stan models.
A small library for Consistent Optimization of Label-wise Utilities in Multi-label clasifficatioN