Links for jaxtorch