Remove JAX 0.7.1 from the unittest matrix and bump the min requirement to 0.7.2#182

wstcliyu merged 1 commit intomainAI-Hypercomputer/pathways-utils:mainfrom

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