2025 Python Packaging Survey is now live!  Take the survey now

Probabilistic programming and deep learning in JAX

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for sharadmv from gravatar.com sharadmv Avatar for tfp-automation from gravatar.com tfp-automation Avatar for tfp-team from gravatar.com tfp-team

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: Apache Software License (Apache 2.0)
  • Author: Google LLC
  • Tags jax , probability , statistics , bayesian , machine , learning
  • Requires: Python >=3.6
Classifiers
Report project as malware

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!

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page