Links for jax-cgd