Save the date! Google I/O returns May 18-20 Register now

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.1.4'
config Instance of jax.config.Config