Parametric Inversion

Program transformation technique that inverts non-injective functions by introducing parameters to uniquely determine inputs, implemented in JAX.

Implemented a program transformation technique in JAX that inverts non-injective functions by introducing parameters to uniquely determine inputs. Demonstrated the approach on simulation models including Lotka-Volterra dynamics, enabling efficient conditioning in probabilistic programs.