lukaslinauer

Hi! ### Problem Description I tried to run the examples for the functional_lagrangian extensions (https://github.com/google-deepmind/jax_verify/tree/master/jax_verify/extensions/functional_lagrangian), but the script fails because the `extensions` module was not found. ### Reproduce The precise...