Links for jax-mlx-plugin