Pin JAX version to less than v0.6.0 due to breaking changes#300

matt-graham merged 1 commit intomainastro-informatics/s2fft:mainfrom

mmg/jax-v0.6.0-pinastro-informatics/s2fft:mmg/jax-v0.6.0-pinCopy head branch name to clipboard