ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: oryx.core.interpreters.inverse.rules

Registers inverse and ILDJ rules.

This module also monkey patches jax.nn.sigmoid, jax.scipy.special.logit, and jax.scipy.special.expit to have custom inverses.