Module: oryx.core.interpreters

Contains function transformations implemented using JAX tracing machinery.

Modules

harvest module: Module for the harvest transformation.

inverse module: Module for the inverse function transformation.

log_prob module: Module for log_prob transformation.

propagate module: Module for the propagate custom Jaxpr interpreter.

unzip module: Module for the unzip function transformation.