Skip to content

Jaxlib release v0.4.29

Compare
Choose a tag to compare
@hawkinsp hawkinsp released this 10 Jun 18:31
· 3666 commits to main since this release
  • Bug fixes

    • Fixed a bug where XLA sharded some concatenation operations incorrectly,
      which manifested as an incorrect output for cumulative reductions (#21403).
    • Fixed a bug where XLA:CPU miscompiled certain matmul fusions
      (openxla/xla#13301).
    • Fixes a compiler crash on GPU (#21396).
  • Deprecations

    • jax.tree.map(f, None, non-None) now emits a DeprecationWarning, and will
      raise an error in a future version of jax. None is only a tree-prefix of
      itself. To preserve the current behavior, you can ask jax.tree.map to
      treat None as a leaf value by writing:
      jax.tree.map(lambda x, y: None if x is None else f(x, y), a, b, is_leaf=lambda x: x is None).