Bump jaxlib from 0.4.8 to 0.4.12

[dependabot]

Bumps jaxlib from 0.4.8 to 0.4.12.

Changelog

Sourced from jaxlib's changelog.

jax 0.4.12 (June 8, 2023)

• Changes

  • Added {class}scipy.spatial.transform.Rotation and {class}scipy.spatial.transform.Slerp

• Deprecations

  • jax.abstract_arrays and its contents are now deprecated. See related functionality in :mod:jax.core.
  • jax.numpy.alltrue: use jax.numpy.all. This follows the deprecation of numpy.alltrue in NumPy version 1.25.0.
  • jax.numpy.sometrue: use jax.numpy.any. This follows the deprecation of numpy.sometrue in NumPy version 1.25.0.
  • jax.numpy.product: use jax.numpy.prod. This follows the deprecation of numpy.product in NumPy version 1.25.0.
  • jax.numpy.cumproduct: use jax.numpy.cumprod. This follows the deprecation of numpy.cumproduct in NumPy version 1.25.0.
  • jax.sharding.OpShardingSharding has been removed since it has been 3 months since it was deprecated.

jaxlib 0.4.12 (June 8, 2023)

• Changes

  • Includes PTX/SASS for Hopper (SM version 9.0+) GPUs. Previous versions of jaxlib should work on Hopper but would have a long JIT-compilation delay the first time a JAX operation was executed.

• Bug fixes

  • Fixes incorrect source line information in JAX-generated Python tracebacks under Python 3.11.
  • Fixes crash when printing local variables of frames in JAX-generated Python tracebacks (#16027).

jax 0.4.11 (May 31, 2023)

• Deprecations
  • The following APIs have been removed after a 3 month deprecation period, in accordance with the {ref}api-compatibility policy:
    • jax.experimental.PartitionSpec: use jax.sharding.PartitionSpec.
    • jax.experimental.maps.Mesh: use jax.sharding.Mesh
    • jax.experimental.pjit.NamedSharding: use jax.sharding.NamedSharding.
    • jax.experimental.pjit.PartitionSpec: use jax.sharding.PartitionSpec.
    • jax.experimental.pjit.FROM_GDA. Instead pass sharded jax.Array objects as input and remove the optional in_shardings argument to pjit.
    • jax.interpreters.pxla.PartitionSpec: use jax.sharding.PartitionSpec.
    • jax.interpreters.pxla.Mesh: use jax.sharding.Mesh
    • jax.interpreters.xla.Buffer: use jax.Array.
    • jax.interpreters.xla.Device: use jax.Device.
    • jax.interpreters.xla.DeviceArray: use jax.Array.
    • jax.interpreters.xla.device_put: use jax.device_put.
    • jax.interpreters.xla.xla_call_p: use jax.experimental.pjit.pjit_p.

... (truncated)

Commits

• 4c02f2c Merge pull request #16305 from skye:version
• 9dccaad Remove unnecessary assignment to num_tokens
• 8223e45 Update WORKSPACE and setup.py for jax + jaxlib 0.4.12 release
• 1445149 Delete OpShardingSharding export since it has been 3 months since it was depr...
• 03b06b6 Make result_shardings's documentation more accurate
• 6374b73 [XLA:Python] Fix crash when accessing locals of JAX-generated Python tracebacks.
• a47aca8 [XLA:Python] Fix incorrect code source information under Python 3.11.
• e598bc8 [PJRT PLUGIN] Pass in a blocking key value get and a key value put function p...
• 5c5bacc Merge pull request #16072 from axch:stacked-piles
• 237625e Merge pull request #16276 from skye:maxfail
• 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)

[dependabot]

Superseded by #59.