compmem / numpyro Goto Github PK
View Code? Open in Web Editor NEWThis project forked from pyro-ppl/numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Home Page: https://num.pyro.ai
License: Apache License 2.0