Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 26, 2025

• Merge pull request #29630 from carlosgmartin:scan_unroll_zero

  Show description for 21027c6
  Google-ML-Automation

  committed
  21027c6

  Copy full SHA for 21027c6

Commits on Oct 25, 2025

• Update XLA dependency to use revision http://github.com/openxla/xla/commit/

  Show description for 11befd4
  Google-ML-Automation

  committed
  11befd4

  Copy full SHA for 11befd4

• Reverts eeaa8050f7488b580e5667d43360c98f71b22a05

  Show description for adbe62b
  Google-ML-Automation

  committed
  adbe62b

  Copy full SHA for adbe62b

Commits on Oct 24, 2025

• [mutable-arrays] enable vjp3 by default

  Show description for eeaa805
  

  mattjj

  authored and

  Google-ML-Automation

  committed
  eeaa805

  Copy full SHA for eeaa805

• [mutable-arrays] in vjp3, don't stuff GradValue sentinels into user pytree nodes

  Show description for cb69cab
  

  2 peopleauthored and

  Google-ML-Automation

  committed
  cb69cab

  Copy full SHA for cb69cab

• [Pallas/TPU] Add supports_sparse_core property to TpuInfo.

  Show description for e4292e6
  Google-ML-Automation

  committed
  e4292e6

  Copy full SHA for e4292e6

• Integrate LLVM at llvm/llvm-project@917d1f20aecf

  Show description for 9366503
  Google-ML-Automation

  committed
  9366503

  Copy full SHA for 9366503

• [pallas:sc] Error on direct loads/stores from/to HBM and VMEM_SHARED

  Show description for b815692
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  b815692

  Copy full SHA for b815692

• Merge pull request #32853 from mattjj:fix-pjit-transpose-bug

  Show description for 15e9723
  Google-ML-Automation

  committed
  15e9723

  Copy full SHA for 15e9723

• Update XLA dependency to use revision http://github.com/openxla/xla/commit/199cdf5c08d142733437b29acf0efba738700159

  Show description for ea61015
  Google-ML-Automation

  committed
  ea61015

  Copy full SHA for ea61015

Commits on Oct 23, 2025

• [mutable-arrays] fix vjp3 fancy transpose bug

  mattjj

  committed
  8c5f4a3

  Copy full SHA for 8c5f4a3

• Merge pull request #29360 from ravwojdyla:rav-fixup-doc

  Show description for bc5fc36
  Google-ML-Automation

  committed
  bc5fc36

  Copy full SHA for bc5fc36

• [MOSAIC] Use shape invariant numerics by default and opt into shape variant numerics.

  Show description for 7a5c5a1
  

  blakehechtman

  authored and

  Google-ML-Automation

  committed
  7a5c5a1

  Copy full SHA for 7a5c5a1

• [JAX] Add compiled program call_location logging support to JAX

  Show description for 9a61e09
  Google-ML-Automation

  committed
  9a61e09

  Copy full SHA for 9a61e09

• Error out if unreduced or reduced is an input to any collective.

  Show description for 4e5a9af
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  4e5a9af

  Copy full SHA for 4e5a9af

• Ensure that the weakref cache's LRU list outlives the LRU caches that use it.

  Show description for 8d58e7c
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  8d58e7c

  Copy full SHA for 8d58e7c

• Added shutdown timeout to `jax.distributed.initialize`.

  Show description for b93bb7d
  

  mwhittaker

  authored and

  Google-ML-Automation

  committed
  b93bb7d

  Copy full SHA for b93bb7d

• [pallas:sc] Added a test for lax.split

  Show description for 35cde24
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  35cde24

  Copy full SHA for 35cde24

• Merge pull request #28997 from NeilGirdhar:annotation

  Show description for 1467437
  Google-ML-Automation

  committed
  1467437

  Copy full SHA for 1467437

• Fixup Potential solution 2 Custom pytree init

  ravwojdyla

  committed
  0dffef2

  Copy full SHA for 0dffef2

• Merge pull request #32844 from jakevdp:version-support

  Show description for 1160b89
  Google-ML-Automation

  committed
  1160b89

  Copy full SHA for 1160b89

• Tighten numpy.linalg.qr annotation

  NeilGirdhar

  committed
  702bd8c

  Copy full SHA for 702bd8c

• [pmap] Make sure `cpp_pmap` gets tested as well, if only internally, until we fully deprecate.

  Show description for 3dd28ce
  

  danielsuo

  authored and

  Google-ML-Automation

  committed
  3dd28ce

  Copy full SHA for 3dd28ce

• [doc] update python/numpy/scipy version support table

  jakevdp

  committed
  ab8e296

  Copy full SHA for ab8e296

• Manually imported https://github.com/jax-ml/jax/pull/20791

  Show description for 87bccb7
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  87bccb7

  Copy full SHA for 87bccb7

• [docs] Manually imported https://github.com/jax-ml/jax/pull/20549

  Show description for 50f48ed
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  50f48ed

  Copy full SHA for 50f48ed

• Disable ASAN for a gpu test that's failing due to a memory leak in NCCL.

  Show description for 72c445a
  

  emilyfertig

  authored and

  Google-ML-Automation

  committed
  72c445a

  Copy full SHA for 72c445a

• Reverts 36dd04ebd317133b1127faa2ff211291494e3ee7

  Show description for 870aa9b
  Google-ML-Automation

  committed
  870aa9b

  Copy full SHA for 870aa9b

• [pallas:mosaic] Added type annotations for a few primitives

  Show description for fe82287
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  fe82287

  Copy full SHA for fe82287

• Merge pull request #32830 from superbobry:main

  Show description for fe76261
  Google-ML-Automation

  committed
  fe76261

  Copy full SHA for fe76261

• Fixed the jupytext failure at head

  superbobry

  committed
  49d4367

  Copy full SHA for 49d4367

• Merge pull request #24350 from jakevdp:eig-grad-error

  Show description for bfed503
  Google-ML-Automation

  committed
  bfed503

  Copy full SHA for bfed503

• Merge pull request #24136 from jjyyxx:patch-3

  Show description for 3be154c
  Google-ML-Automation

  committed
  3be154c

  Copy full SHA for 3be154c

• Merge pull request #26682 from alhridoy:docs-debug-print-perturbation

  Show description for 38637b3
  Google-ML-Automation

  committed
  38637b3

  Copy full SHA for 38637b3

• Merge pull request #27292 from uuirs:fix_decode_attention

  Show description for 5d5f744
  Google-ML-Automation

  committed
  5d5f744

  Copy full SHA for 5d5f744

Pagination

Previous

Next