Have a question? Connect with the community at the TensorFlow Forum Visit Forum

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.