Relax JAX constraint from jax<0.7 to jax#107

epignatelli merged 3 commits intomainepignatelli/navix:mainfrom

copilot/fix-104epignatelli/navix:copilot/fix-104Copy head branch name to clipboard