Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Why do you think that https://github.com/arogozhnikov/einops is a good alternative to numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Why do you think that https://github.com/arogozhnikov/einops is a good alternative to numpyro