Stars
2
stars
written in Python
Clear filter
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Oryx is a library for probabilistic programming and deep learning built on top of Jax.