Links for jax-optix