Using bilevel optimization to add an optimal transport-based metric to the birdflow model objective, planned research for Spring 2025. Contains initial tests of ott-jax and jaxopt, and a jax/optax implementation of bilevel optimization for a toy example!