Oryx یک کتابخانه برای برنامه نویسی احتمالی و یادگیری عمیق است که بر روی JAX ساخته شده است.
import oryx import jax.numpy as jnp ppl = oryx.core.ppl tfd = oryx.distributions # Define sampling function def sample(key): x = ppl.random_variable(tfd.Normal(0., 1.))(key) return jnp.exp(x / 2.) + 2. # Transform sampling function into a log-density function ppl.log_prob(sample)(1.) # ==> -0.9189
رویکرد Oryx نمایش مجموعهای از تبدیلهای تابع است که تبدیلهای موجود JAX را ترکیب کرده و با آنها ادغام میکند. برای نصب Oryx می توانید اجرا کنید:
pip install --upgrade oryx