Module: oryx.core.interpreters.inverse.bijector_extensions

Wraps TFP bijectors for use with Jax.

Functions

patch_bijector(...): Patches a TFP bijector to use a primitive in its forward/inverse methods.