Bump jax from 0.8.1 to 0.8.2

[dependabot]

Bumps jax from 0.8.1 to 0.8.2.

Release notes

Sourced from jax's releases.

JAX v0.8.2

• Deprecations

  • jax.lax.pvary has been deprecated. Please use jax.lax.pcast(..., to='varying') as the replacement.
  • Complex arguments passed to jax.numpy.arange now result in a deprecation warning, because the output is poorly-defined.
  • From jax.core a number of symbols are newly deprecated including: call_impl, get_aval, mapped_aval, subjaxprs, set_current_trace, take_current_trace, traverse_jaxpr_params, unmapped_aval, AbstractToken, and TraceTag.
  • All symbols in jax.interpreters.pxla are deprecated. These are primarily JAX internal APIs, and users should not rely on them.

• Changes:

  • jax's Tracer no longer inherits from jax.Array at runtime. However, jax.Array now uses a custom metaclass such isinstance(x, Array) is true if an object x represents a traced Array. Only some Tracers represent Arrays, so it is not correct for Tracer to inherit from Array.

    For the moment, during Python type checking, we continue to declare Tracer as a subclass of Array, however we expect to remove this in a future release.

  • jax.experimental.si_vjp has been deleted. jax.vjp subsumes it's functionality.

Changelog

Sourced from jax's changelog.

JAX 0.8.2 (December 18, 2025)

• Deprecations

  • jax.lax.pvary has been deprecated. Please use jax.lax.pcast(..., to='varying') as the replacement.
  • Complex arguments passed to {func}jax.numpy.arange now result in a deprecation warning, because the output is poorly-defined.
  • From {mod}jax.core a number of symbols are newly deprecated including: call_impl, get_aval, mapped_aval, subjaxprs, set_current_trace, take_current_trace, traverse_jaxpr_params, unmapped_aval, AbstractToken, and TraceTag.
  • All symbols in {mod}jax.interpreters.pxla are deprecated. These are primarily JAX internal APIs, and users should not rely on them.

• Changes:

  • jax's Tracer no longer inherits from jax.Array at runtime. However, jax.Array now uses a custom metaclass such isinstance(x, Array) is true if an object x represents a traced Array. Only some Tracers represent Arrays, so it is not correct for Tracer to inherit from Array.

    For the moment, during Python type checking, we continue to declare Tracer as a subclass of Array, however we expect to remove this in a future release.

  • jax.experimental.si_vjp has been deleted. jax.vjp subsumes it's functionality.

Commits

• 3f12502 Prepare for JAX release 0.8.2
• e2815d5 [pallas:mosaic] pltpu.emit_pipeline now accepts block specs in HBM
• 69a6cca [mpmd] Fix stage_id field nanobind compatability
• 315bb93 [MGPU] Doc fix and shape size fix for broadcast in WGSplatFragLayout.
• c1188e7 Add getitem to backwards compatible shims.
• b9b2d4f [pallas:mosaic] Recover memory_space from the aval in aval_to_ir_type
• 3349e7a [Pallas] Allow inferring the backend from the provided compiler params.
• 97c557d Update XLA dependency to use revision http://github.com/openxla/xla/commit/91...
• 286e358 Automated Code Change
• f81c201 [pallas:mosaic] Access memory spaces via pltpu directly
• 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 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)

[github-actions]

✅MegaLinter analysis: Success

Descriptor	Linter	Files	Fixed	Errors	Warnings	Elapsed time
✅ COPYPASTE	jscpd	yes		no	no	5.96s
✅ JSON	prettier	2	0	0	0	0.41s
✅ JSON	v8r	2		0	0	2.82s
✅ MARKDOWN	markdownlint	1	0	0	0	0.64s
✅ MARKDOWN	markdown-table-formatter	1	0	0	0	0.25s
✅ PYTHON	bandit	247		0	0	3.24s
✅ PYTHON	black	247	0	0	0	5.15s
✅ PYTHON	flake8	247		0	0	1.71s
✅ PYTHON	isort	247	0	0	0	0.48s
✅ PYTHON	mypy	247		0	0	3.86s
✅ PYTHON	pylint	247		0	0	62.62s
✅ PYTHON	ruff	247	0	0	0	0.03s
✅ REPOSITORY	checkov	yes		no	no	14.52s
✅ REPOSITORY	gitleaks	yes		no	no	2.41s
✅ REPOSITORY	git_diff	yes		no	no	0.0s
✅ REPOSITORY	secretlint	yes		no	no	3.27s
✅ REPOSITORY	syft	yes		no	no	1.3s
✅ REPOSITORY	trivy-sbom	yes		no	no	1.04s
✅ REPOSITORY	trufflehog	yes		no	no	10.71s
✅ YAML	prettier	4	0	0	0	0.47s
✅ YAML	v8r	4		0	0	4.85s
✅ YAML	yamllint	4		0	0	0.39s

See detailed reports in MegaLinter artifacts

Your project could benefit from a custom flavor, which would allow you to run only the linters you need, and thus improve runtime performances. (Skip this info by defining FLAVOR_SUGGESTIONS: false)

• Documentation: Custom Flavors
• Command: npx mega-linter-runner@9.2.0 --custom-flavor-setup --custom-flavor-linters PYTHON_PYLINT,PYTHON_BLACK,PYTHON_FLAKE8,PYTHON_ISORT,PYTHON_BANDIT,PYTHON_MYPY,PYTHON_RUFF,COPYPASTE_JSCPD,JSON_V8R,JSON_PRETTIER,MARKDOWN_MARKDOWNLINT,MARKDOWN_MARKDOWN_TABLE_FORMATTER,REPOSITORY_CHECKOV,REPOSITORY_GIT_DIFF,REPOSITORY_GITLEAKS,REPOSITORY_SECRETLINT,REPOSITORY_SYFT,REPOSITORY_TRIVY_SBOM,REPOSITORY_TRUFFLEHOG,YAML_PRETTIER,YAML_YAMLLINT,YAML_V8R