https://github.com/jax-ml/oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
https://github.com/jax-ml/oryx
jax probabilistic-programming python
Last synced: 10 months ago
JSON representation
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
- Host: GitHub
- URL: https://github.com/jax-ml/oryx
- Owner: jax-ml
- License: apache-2.0
- Created: 2022-07-11T21:13:01.000Z (over 3 years ago)
- Default Branch: main
- Last Pushed: 2025-03-28T00:12:48.000Z (11 months ago)
- Last Synced: 2025-04-03T17:11:14.395Z (10 months ago)
- Topics: jax, probabilistic-programming, python
- Language: Python
- Homepage: https://tensorflow.org/probability/oryx
- Size: 357 KB
- Stars: 252
- Watchers: 10
- Forks: 11
- Open Issues: 17
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
Awesome Lists containing this project
README
# 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`).
*This is not an official Google product*
## Installation
You can install Oryx via `pip`:
```bash
$ pip install oryx
```
## Documentation and Examples
Documentation is available
[on the Oryx website](https://www.tensorflow.org/probability/oryx/api_docs/python/oryx).
### Guides
* [A tour of Oryx](https://www.tensorflow.org/probability/oryx/notebooks/a_tour_of_oryx)
* [Probabilistic programming](https://www.tensorflow.org/probability/oryx/notebooks/probabilistic_programming)
## Development
To develop and modify Oryx, you need to install
[`poetry`](https://python-poetry.org/), a tool for Python packaging and
dependency management.
To install the development dependencies of Oryx, you can run
```bash
$ poetry install
```
and to enter a virtual environment for testing or debugging, you can run:
```bash
$ poetry shell
```
### Running tests
Oryx uses [Pytest](https://pytest.org/) for testing. To run the tests, use the following command:
```
$ poetry run pytest oryx
```