chore(deps-dev): update jaxlib requirement from <=0.1.65 to <=0.4.7

[dependabot[bot]]

Updates the requirements on jaxlib to permit the latest version.

Changelog

Sourced from jaxlib's changelog.

jax 0.4.7 (March 27, 2023)

• Changes

  • As per https://jax.readthedocs.io/en/latest/jax_array_migration.html#jax-array-migration jax.config.jax_array cannot be disabled anymore.
  • jax.config.jax_jit_pjit_api_merge cannot be disabled anymore.
  • {func}jax.experimental.jax2tf.convert now supports the native_serialization parameter to use JAX's native lowering to StableHLO to obtain a StableHLO module for the entire JAX function instead of lowering each JAX primitive to a TensorFlow op. This simplifies the internals and increases the confidence that what you serialize matches the JAX native semantics. See documentation. As part of this change the config flag --jax2tf_default_experimental_native_lowering has been renamed to --jax2tf_native_serialization.
  • JAX now depends on ml_dtypes, which contains definitions of NumPy types like bfloat16. These definitions were previously internal to JAX, but have been split into a separate package to facilitate sharing them with other projects.
  • JAX now requires NumPy 1.21 or newer and SciPy 1.7 or newer.

• Deprecations

  • The type jax.numpy.DeviceArray is deprecated. Use jax.Array instead, for which it is an alias.
  • The type jax.interpreters.pxla.ShardedDeviceArray is deprecated. Use jax.Array instead.
  • Passing additional arguments to {func}jax.numpy.ndarray.at by position is deprecated. For example, instead of x.at[i].get(True), use x.at[i].get(indices_are_sorted=True)
  • jax.interpreters.xla.device_put is deprecated. Please use jax.device_put.
  • jax.interpreters.pxla.device_put is deprecated. Please use jax.device_put.
  • jax.experimental.pjit.FROM_GDA is deprecated. Please pass in sharded jax.Arrays as input and remove the in_shardings argument to pjit since it is optional.

jaxlib 0.4.7 (March 27, 2023)

Changes:

• jaxlib now depends on ml_dtypes, which contains definitions of NumPy types like bfloat16. These definitions were previously internal to JAX, but have been split into a separate package to facilitate sharing them with other projects.

jax 0.4.6 (Mar 9, 2023)

• Changes
  • jax.tree_util now contain a set of APIs that allow user to define keys for their custom pytree node. This includes:
    • tree_flatten_with_path that flattens a tree and return not only each leaf but also their key paths.
    • tree_map_with_paths that can map a function that takes the key path as argument.
    • `register_pytree_with_keys`` to register how the key path and leaves should looks

... (truncated)

Commits

• ae4f1fc Update the commit in workspace too
• e9cac5e Prepare for jax and jaxlib 0.4.7 release
• e21aee1 Add deprecation warning for FROM_GDA usage since that argument is not require...
• 3c3fa04 Copy seq_lengths before creating descriptor
• 88c2898 Use pytype_strict_library() in Bazel build rules.
• 40fb646 Fix duplicate definition of 'cuda' extra in setup.py.
• af4d494 Merge pull request #8633 from shawwn:2021-11-19/autodidax-fix-jaxpr-subcomp-r...
• 10d51c7 Merge pull request #15218 from hawkinsp:mypy
• 41695cc Temporarily fix the compilation cache test which is failing on latest jaxlib ...
• 2c4be6f Merge pull request #15226 from canyon289:patch-1
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR: - `@dependabot rebase` will rebase this PR - `@dependabot recreate` will recreate this PR, overwriting any edits that have been made to it - `@dependabot merge` will merge this PR after your CI passes on it - `@dependabot squash and merge` will squash and merge this PR after your CI passes on it - `@dependabot cancel merge` will cancel a previously requested merge and block automerging - `@dependabot reopen` will reopen this PR if it is closed - `@dependabot close` will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually - `@dependabot ignore this major version` will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself) - `@dependabot ignore this minor version` will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself) - `@dependabot ignore this dependency` will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)