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.