Remove JAX 0.7.2 from the unittest matrix and bump the min requirement to 0.8.0#206

wstcliyu wants to merge 2 commits intomainAI-Hypercomputer/pathways-utils:mainfrom

wstcliyu/remove-jax-072AI-Hypercomputer/pathways-utils:wstcliyu/remove-jax-072Copy head branch name to clipboard