chore(deps): update dependency jaxlib to v0.4.38 #273
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR contains the following updates:
==0.4.36.dev20241122
->==0.4.38
Release Notes
jax-ml/jax (jaxlib)
v0.4.38
Changes:
jax.tree.flatten_with_path
andjax.tree.map_with_path
are addedas shortcuts of the corresponding
tree_util
functions.Deprecations
jax.core
namespace have been deprecated.Most were no-ops, were little-used, or can be replaced by APIs of the same
name in {mod}
jax.extend.core
; see the documentation for {mod}jax.extend
for information on the compatibility guarantees of these semi-public extensions.
jax.core
:check_eqn
,check_type
,check_valid_jaxtype
, andnon_negative_dim
.jax.lib.xla_bridge
:xla_client
anddefault_backend
.jax.lib.xla_client
:_xla
andbfloat16
.jax.numpy
:round_
.New Features
jax.export.export
can be used for device-polymorphic export withshardings constructed with {func}
jax.sharding.AbstractMesh
.See the jax.export documentation.
jax.lax.split
. This is a primitive version of{func}
jax.numpy.split
, added because it yields a more compacttranspose during automatic differentiation.
Configuration
📅 Schedule: Branch creation - "after 5am on saturday" (UTC), Automerge - At any time (no schedule defined).
🚦 Automerge: Disabled by config. Please merge this manually once you are satisfied.
♻ Rebasing: Whenever PR becomes conflicted, or you tick the rebase/retry checkbox.
🔕 Ignore: Close this PR and you won't be reminded about this update again.
To execute skipped test pipelines write comment
/ok-to-test
.This PR has been generated by MintMaker (powered by Renovate Bot).