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!
Project details
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.2.dev0.tar.gz
(90.8 kB
view hashes)