Pyro PPL on NumPy
conda install sfe1ed40::numpyro
Probabilistic programming powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.