Convert torchtyping to jaxtyping for PyTorch>=2.4#7

iwyoo wants to merge 1 commit intochython:masterchython/chytorch:masterfrom

iwyoo:chore/jaxtyping-migrationiwyoo/chytorch:chore/jaxtyping-migrationCopy head branch name to clipboard