Links for jax-bitempered-loss