Bumps JAX version to 0.2.13

Commit

4 years ago

Bumps JAX version to 0.2.13 This includes several new features, like for example the `jax.host_id()` -> `jax.process_index()` renaming.