semvec/pypi-packages huggingface dataset:
oryx 0.1.4
Probabilistic programming and deep learning in JAX
License: Apache 2.0
Requires Python: >=3.6
Keywords: jax probability statistics bayesian machine learning

# Oryx

Oryx is a library for probabilistic programming and deep learning built on top
of Jax. The approach is to expose a set of function transformations that
compose and integrate with JAX's existing transformations (e.g. `jit`, `grad`,
and `vmap`).

## Documentation and Examples

Coming soon!