Bump the python group with 4 updates

[dependabot]

Updates the requirements on tensorflow-text, torch, torchvision and jax[cuda12_pip] to permit the latest version.
Updates tensorflow-text to 2.16.1

Release notes

Sourced from tensorflow-text's releases.

v2.16.1

Release 2.16.1

Major Features and Improvements

Breaking Changes

Bug Fixes and Other Changes

• Update tf-text setup scripts.
• Support resource manager scoped Sentencepiece resources.
• Remove use_unique_shared_resource_name.
• Remove tensorflow_text dependency on tf_hub library.
• Fix TF patch, bump TF commit
• Update version

Thanks to our Contributors

This release contains contributions from many people at Google, as well as:

Raviteja Gorijala

Commits

• See full diff in compare view

Updates torch from 2.1.2 to 2.2.1+cu121

Updates torchvision from 0.16.2 to 0.17.1+cu121

Updates jax[cuda12_pip] from 0.4.23 to 0.4.25

Release notes

Sourced from jax[cuda12_pip]'s releases.

JAX release v0.4.25

No release notes provided.

Jaxlib release v0.4.25

No release notes provided.

JAX release v0.4.24

jaxlib release v0.4.24

Changelog

Sourced from jax[cuda12_pip]'s changelog.

jax 0.4.25 (Feb 26, 2024)

• New Features

  • Added CUDA Array Interface import support (requires jaxlib 0.4.24).
  • JAX arrays now support NumPy-style scalar boolean indexing, e.g. x[True] or x[False].
  • Added {mod}jax.tree module, with a more convenient interface for referencing functions in {mod}jax.tree_util.
  • {func}jax.tree.transpose (i.e. {func}jax.tree_util.tree_transpose) now accepts inner_treedef=None, in which case the inner treedef will be automatically inferred.

• Changes

  • Pallas now uses XLA instead of the Triton Python APIs to compile Triton kernels. You can revert to the old behavior by setting the JAX_TRITON_COMPILE_VIA_XLA environment variable to "0".
  • Several deprecated APIs in {mod}jax.interpreters.xla that were removed in v0.4.24 have been re-added in v0.4.25, including backend_specific_translations, translations, register_translation, xla_destructure, TranslationRule, TranslationContext, and XLAOp. These are still considered deprecated, and will be removed again in the future when better replacements are available. Refer to {jax-issue}[#19816](https://github.com/google/jax/issues/19816) for discussion.

• Deprecations & Removals

  • {func}jax.numpy.linalg.solve now shows a deprecation warning for batched 1D solves with b.ndim > 1. In the future these will be treated as batched 2D solves.
  • Conversion of a non-scalar array to a Python scalar now raises an error, regardless of the size of the array. Previously a deprecation warning was raised in the case of non-scalar arrays of size 1. This follows a similar deprecation in NumPy.
  • The previously deprecated configuration APIs have been removed following a standard 3 months deprecation cycle (see {ref}api-compatibility). These include
    • the jax.config.config object and
    • the define_*_state and DEFINE_* methods of {data}jax.config.
  • Importing the jax.config submodule via import jax.config is deprecated. To configure JAX use import jax and then reference the config object via jax.config.
  • The minimum jaxlib version is now 0.4.20.

jaxlib 0.4.25 (Feb 26, 2024)

jax 0.4.24 (Feb 6, 2024)

• Changes

  • JAX lowering to StableHLO does not depend on physical devices anymore. If your primitive wraps custom_paritioning or JAX callbacks in the lowering rule i.e. function passed to rule parameter of mlir.register_lowering then add your primitive to jax._src.dispatch.prim_requires_devices_during_lowering set.

... (truncated)

Commits

• f9e20d5 Start the release of jax and jaxlib 0.4.25
• f4045dc Remove the deprecation of jax.tree_map for the release of 0.4.25
• 22996bb Update XLA dependency to use revision
• 61aa7e8 [Mosaic] Fix bug in divisibility check in infer_vector_layout load and store ...
• 072b43b Merge pull request #19960 from mattjj:attrs-autodiff
• b0b88d8 [attrs] add linearize and vjp support
• ae9d677 Merge pull request #19936 from tillahoffmann:fix-19935
• e1e9de0 Merge pull request #19499 from nstarman:array_api-broadcast_to-type-hint
• 75cdef7 [Mosaic][NFC] Prefer mlir aliases for llvm classes/functions within mlir name...
• 8a43140 [Mosaic][apply_vector_layout][NFC] Use LLVM_UNLIKELY in TPU_ASSERT_* macros
• 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 show <dependency name> ignore conditions will show all of the ignore conditions of the specified dependency
• @dependabot ignore <dependency name> major version will close this group update PR and stop Dependabot creating any more for the specific dependency's major version (unless you unignore this specific dependency's major version or upgrade to it yourself)
• @dependabot ignore <dependency name> minor version will close this group update PR and stop Dependabot creating any more for the specific dependency's minor version (unless you unignore this specific dependency's minor version or upgrade to it yourself)
• @dependabot ignore <dependency name> will close this group update PR and stop Dependabot creating any more for the specific dependency (unless you unignore this specific dependency or upgrade to it yourself)
• @dependabot unignore <dependency name> will remove all of the ignore conditions of the specified dependency
• @dependabot unignore <dependency name> <ignore condition> will remove the ignore condition of the specified dependency and ignore conditions

[mattdangerw]

@sampathweb any thoughts on the jax version bump? Should we be excluding jax/torch/tf version from dependabot or still helpful?

[sampathweb]

Don't want to update Jax. Created PR: #1507 to disable autoupdate of Jax.

[sampathweb]

@sampathweb any thoughts on the jax version bump? Should we be excluding jax/torch/tf version from dependabot or still helpful?

Yeah. I just now created PR #1507 to do that. Not sure about Torch or TF Version. Should we add them as well? So far they seem to be compatible with the CI.

[dependabot]

Looks like these dependencies are no longer being updated by Dependabot, so this is no longer needed.