Parametric Inversion
Program transformation technique that inverts non-injective functions by introducing parameters to uniquely determine inputs, implemented in JAX.
Implemented the parametric inversion framework introduced by Zenna Tavares and Armando Solar-Lezama in Parametric Inverse Simulation. The JAX implementation inverts non-injective functions by introducing parameters to uniquely determine inputs. Demonstrated the approach on simulation models including Lotka-Volterra dynamics, enabling efficient sampling of posteriors.