Commits

Branch selector

 cachemiss

User selector

All users

Datepicker

All time

Commit History

Commits on Sep 16, 2025

• Fix a concurrency problem in jax_explain_cache_misses.

  Show description for 0e352de
  hawkinsp

  committed
  0e352de

  Copy full SHA for 0e352de

• Remove dead code after jax 0.7.2 release. jaxlib_extension_version is 373 and ifrt_version is 27

  Show description for 36a716e
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  36a716e

  Copy full SHA for 36a716e

• Merge pull request #31844 from jax-ml:postrelease-0.7.2

  Show description for 5d35b6a
  Google-ML-Automation

  committed
  5d35b6a

  Copy full SHA for 5d35b6a

• [Mosaic:TPU][Relayouts] Specialized kNone -> kSecondMinor with replicated 2nd minor relayout

  Show description for 6ebf175
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  6ebf175

  Copy full SHA for 6ebf175

• Disable `LaxBackedNumpyTests.testUnravelIndex` under debug mode build.

  Show description for 498e493
  

  danielsuo

  authored and

  Google-ML-Automation

  committed
  498e493

  Copy full SHA for 498e493

• Merge pull request #31848 from kanglant:lock_updates_after_0.7.2

  Show description for 9bede50
  Google-ML-Automation

  committed
  9bede50

  Copy full SHA for 9bede50

• [pallas:mosaic] Removed `MosaicGridMapping.get_extra_args`

  Show description for 3aaa658
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  3aaa658

  Copy full SHA for 3aaa658

• Update jax artifacts and libtpu versions in the requirements lock files

  kanglant

  committed
  5710f31

  Copy full SHA for 5710f31

• Merge pull request #31843 from hawkinsp:install

  Show description for 39d0391
  Google-ML-Automation

  committed
  39d0391

  Copy full SHA for 39d0391

• Add assertion to prevent full manual shardings in `get_num_ways_dim_sharded` and remove the wsc in debug_print lowering

  Show description for a973f84
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  a973f84

  Copy full SHA for a973f84

• Merge remote-tracking branch 'origin/release/0.7.2' into postrelease/0.7.2

  kanglant

  committed
  398488c

  Copy full SHA for 398488c

• Add CUDA 13 to the installation documentation.

  hawkinsp

  committed
  287fd89

  Copy full SHA for 287fd89

• [pmap] Reset stores for WrappedFun passed to _cached_shard_map when implementing pmap via shard_map.

  Show description for e137e2e
  

  danielsuo

  authored and

  Google-ML-Automation

  committed
  e137e2e

  Copy full SHA for e137e2e

• [Mosaic] Allow 3rd minor <-> 2nd minor transpose with non-sublane aligned dim.

  Show description for 9162b36
  

  WindQAQ

  authored and

  Google-ML-Automation

  committed
  9162b36

  Copy full SHA for 9162b36

• Remove a canonicalize_value call from partial_eval.py.

  Show description for 5932df5
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  5932df5

  Copy full SHA for 5932df5

• Merge pull request #31809 from vfdev-5:replaced-spec-chars-checkpointing-docs

  Show description for c850bef
  Google-ML-Automation

  committed
  c850bef

  Copy full SHA for c850bef

• Implement pickle/copy for Literal... scalar types.

  Show description for 565b9af
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  565b9af

  Copy full SHA for 565b9af

• Update flatten_axes to avoid jax.tree.unflatten on sentinel objects.

  Show description for 0303f4a
  

  jburnim

  authored and

  Google-ML-Automation

  committed
  0303f4a

  Copy full SHA for 0303f4a

• Fix cuda13 release wheel installation verification tests

  Show description for a4fa882
  

  kanglant

  authored and

  Google-ML-Automation

  committed
  a4fa882

  Copy full SHA for a4fa882

• Suppress a warning seen in Mac CI for testSqrtmPSDMatrix.

  Show description for a174b11
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  a174b11

  Copy full SHA for a174b11

• [multihost_utils] Add PRNGKeyArray support to `host_local_array_to_global_array` and `global_array_to_host_local_array`.

  Show description for 2bb5ce5
  

  danielsuo

  authored and

  Google-ML-Automation

  committed
  2bb5ce5

  Copy full SHA for 2bb5ce5

• Replaced &#151; with --- in remat and checkpointing docs

  vfdev-5

  committed
  9db4bb5

  Copy full SHA for 9db4bb5

• Integrate LLVM at llvm/llvm-project@a1de9aca1150

  Show description for 0ee8240
  Google-ML-Automation

  committed
  0ee8240

  Copy full SHA for 0ee8240

• Add support for iota to SC lowering.

  Show description for f36e496
  

  brianwa84

  authored and

  Google-ML-Automation

  committed
  f36e496

  Copy full SHA for f36e496

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

  Show description for b935509
  Google-ML-Automation

  committed
  b935509

  Copy full SHA for b935509

• [Mosaic GPU] Expose clusterlaunchcontrols in mosaic GPU.

  Show description for 6d4c75f
  

  Rifur13

  authored and

  Google-ML-Automation

  committed
  6d4c75f

  Copy full SHA for 6d4c75f

• [Mosaic:TPU][Relayouts] Combine-halves retilings

  Show description for bd354ec
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  bd354ec

  Copy full SHA for bd354ec

• Fix wsc's abstract_eval rule to update the shardings correctly when the dst_sharding is not a NamedSharding

  Show description for 9e74132
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  9e74132

  Copy full SHA for 9e74132

Commits on Sep 15, 2025

• Prepare for JAX release 0.7.2

  mattjj

  committed
  9423314

  Copy full SHA for 9423314

• Fix device_put's abstract_eval to return the correct sharding on the type when the sharding passed to it changes. Fixes https://github.com/jax-ml/jax/issues/31793

  Show description for 6093c04
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  6093c04

  Copy full SHA for 6093c04

• [Pallas][Mosaic GPU] Improve ragged dot kernel for blackwell.

  Show description for a29751e
  

  justinjfu

  authored and

  Google-ML-Automation

  committed
  a29751e

  Copy full SHA for a29751e

• Merge pull request #31790 from mattjj:hijax-refs

  Show description for 8084f44
  Google-ML-Automation

  committed
  8084f44

  Copy full SHA for 8084f44

• [hijax] support refs to hijax types

  Show description for 1f75be0
  

  mattjj

  and

  dougalm

  committed
  1f75be0

  Copy full SHA for 1f75be0

• [Pallas/Hijax] Let MemoryRef be parameterized by a type, instead of forcing it to be ShapedArray

  Show description for 9a238f1
  

  sharadmv

  authored and

  Google-ML-Automation

  committed
  9a238f1

  Copy full SHA for 9a238f1

• [Mosaic] Add tpu.barrier.

  Show description for a27ef01
  

  sashabu

  authored and

  Google-ML-Automation

  committed
  a27ef01

  Copy full SHA for a27ef01

Pagination

Previous

Next