Closed schroederdewitt closed 2 years ago
see JAX changelog https://jax.readthedocs.io/en/latest/changelog.html?highlight=0.2.22#jax-0-2-22-oct-12-2021
Thank you, looks good! I merged the PR.
see JAX changelog https://jax.readthedocs.io/en/latest/changelog.html?highlight=0.2.22#jax-0-2-22-oct-12-2021