Links for jax-lnn