Commits

Branch selector

 master

User selector

hawkinsp

Datepicker

All time

Commit History

Commits on Dec 20, 2024

• Avoid calls to warnings.catch_warnings in JAX core code.

  Show description for 59e5ce2
  hawkinsp

  committed
  59e5ce2

  Copy full SHA for 59e5ce2

• Migrate _mlir Python binding target to nanobind.

  Show description for 0ff3f14
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  0ff3f14

  Copy full SHA for 0ff3f14

Commits on Dec 19, 2024

• Migrate shardy dialect extension to nanobind.

  Show description for 9c3365f
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  9c3365f

  Copy full SHA for 9c3365f

Commits on Dec 18, 2024

• Migrate mhlo dialect extension to nanobind.

  Show description for 3f24dfd
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  3f24dfd

  Copy full SHA for 3f24dfd

• Migrate StableHLO Python extension to nanobind.

  Show description for 3d54d03
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  3d54d03

  Copy full SHA for 3d54d03

• Increase the minimum NumPy version to v1.25.

  Show description for ee45718
  hawkinsp

  committed
  ee45718

  Copy full SHA for ee45718

• Bump xla_extension_version after jaxlib release.

  Show description for 7776982
  hawkinsp

  committed
  7776982

  Copy full SHA for 7776982

Commits on Dec 17, 2024

• Update version numbers after release.

  hawkinsp

  committed
  ff52aed

  Copy full SHA for ff52aed

• Merge branch 'release/0.4.38'

  hawkinsp

  committed
  fd8de7d

  Copy full SHA for fd8de7d

• Enable PJRT compatibility in cloud TPU CI.

  Show description for 71b23ea
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  71b23ea

  Copy full SHA for 71b23ea

• Release v0.4.38

  hawkinsp

  committed
  c4e3f5a

  Copy full SHA for c4e3f5a

• Reverts 525b646

  Show description for 7de9eb2
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  7de9eb2

  Copy full SHA for 7de9eb2

Commits on Dec 16, 2024

• Add Python.h include to fix Windows build.

  Show description for 11e0fdf
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  11e0fdf

  Copy full SHA for 11e0fdf

Commits on Dec 13, 2024

• Migrate JAX MLIR Python dialect extensions to nanobind.

  Show description for 64eae32
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  64eae32

  Copy full SHA for 64eae32

• Use a broadcasted gather in the sort JVP, rather than forming explicit iotas.

  Show description for 0922feb
  hawkinsp

  committed
  0922feb

  Copy full SHA for 0922feb

Commits on Dec 12, 2024

• Relax test tolerance for complex128 pow in lax_test.py.

  Show description for 6548caf
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  6548caf

  Copy full SHA for 6548caf

• Don't monkey-patch functions in test_utils to count events for tests.

  Show description for 62e66b6
  hawkinsp

  committed
  62e66b6

  Copy full SHA for 62e66b6

Commits on Dec 10, 2024

• Remove code in jax2tf for compatibility with TF 2.10 or earlier.

  hawkinsp

  committed
  acae2f0

  Copy full SHA for acae2f0

• Merge branch 'release/0.4.37' into main.

  hawkinsp

  committed
  820f51d

  Copy full SHA for 820f51d

Commits on Dec 9, 2024

• Update versions for v0.4.37 release.

  hawkinsp

  committed
  ffb07cd

  Copy full SHA for ffb07cd

• Remove dead code after minimum jaxlib version bump to v0.4.36.

  Show description for 79318a0
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  79318a0

  Copy full SHA for 79318a0

Commits on Dec 6, 2024

• Set -Werror=mismatched-tags on clang.

  Show description for fac1b1a
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  fac1b1a

  Copy full SHA for fac1b1a

• Bump JAX version after release.

  Show description for ba626fa
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  ba626fa

  Copy full SHA for ba626fa

Commits on Dec 5, 2024

• Merge branch 'release/0.4.36' after release

  hawkinsp

  committed
  ab02bf8

  Copy full SHA for ab02bf8

• JAX release 0.4.36.

  hawkinsp

  committed
  7e6620a

  Copy full SHA for 7e6620a

Commits on Dec 4, 2024

• Disable pgle_test on non-GPU platforms.

  Show description for 681b9c2
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  681b9c2

  Copy full SHA for 681b9c2

• Disable JaxAotTest.test_topology_pjit_serialize on GPU, which fails in CI.

  Show description for bdadc53
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  bdadc53

  Copy full SHA for bdadc53

• Disable backwards compatibility test for Triton IR.

  Show description for 2ac2692
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  2ac2692

  Copy full SHA for 2ac2692

• Fix Windows portability problem in compilation cache test.

  hawkinsp

  committed
  5a25009

  Copy full SHA for 5a25009

Commits on Nov 20, 2024

• Mention stackless in the release notes.

  hawkinsp

  committed
  dfe27a1

  Copy full SHA for dfe27a1

• Fix a bug where constant deduplication used an inappropriate inequality.

  Show description for 867a361
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  867a361

  Copy full SHA for 867a361

Commits on Nov 19, 2024

• Reverts 2075b09

  Show description for 525b646
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  525b646

  Copy full SHA for 525b646

• Add a new API jax.lax.split.

  Show description for 2c80d1a
  hawkinsp

  committed
  2c80d1a

  Copy full SHA for 2c80d1a

• Update jax.scipy.special.gamma and gammasgn to return NaN for negative integer inputs.

  Show description for c5e8ae8
  hawkinsp

  committed
  c5e8ae8

  Copy full SHA for c5e8ae8

Commits on Nov 18, 2024

• Filter custom dtypes by supported_dtypes in _LazyDtypes.

  Show description for f325051
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  f325051

  Copy full SHA for f325051

Pagination

Previous

Next