Probabilistic programming and deep learning in JAX
Project description
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!
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
oryx-0.1.3.tar.gz
(90.7 kB
view hashes)
