bumping jax and equinox version. Also changing the cli interface to n…#119

chamel3 merged 2 commits intomainsandialabs/pancax:mainfrom

bump-jax-versionsandialabs/pancax:bump-jax-versionCopy head branch name to clipboard