Module: oryx

Oryx is a neural network mini-library built on top of Jax.


bijectors module: Module for probability bijectors and related functions.

core module: Contains Oryx's core transformations and functionality.

distributions module: Module for probability distributions and related functions.

experimental module: Module for experimental Oryx libraries.

internal module: Module for internal utilities like testing.

util module: Contains Oryx utility functions.

version module: Define Oryx version information.

version '0.2.2-dev'