Module: oryx.core.interpreters

Stay organized with collections Save and categorize content based on your preferences.

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.