Links for jax-mppi