Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 27, 2025

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

  Show description for 4657a31
  Google-ML-Automation

  committed
  4657a31

  Copy full SHA for 4657a31

Commits on Dec 26, 2025

• Remove CUDA toolchains from CPU-only non-hermetic builds to address https://github.com/jax-ml/jax/issues/33503.

  Show description for 816e644
  

  ybaturina

  authored and

  Google-ML-Automation

  committed
  816e644

  Copy full SHA for 816e644

• Make pvary handle remove_size_one_mesh_axis_from_type. We should remove remove_size_one_mesh_axis_from_type from JAX generally and handle this at the user level.

  Show description for a23c2d3
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  a23c2d3

  Copy full SHA for a23c2d3

• Fix shmap partial_eval_custom_rule which didn't take in_fwd into account while calculating out_res_specs_known when check_vma=True. TODO is to add a test.

  Show description for 714212b
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  714212b

  Copy full SHA for 714212b

Commits on Dec 25, 2025

• [Pallas] Fix broken tests after Mosaic GPU became the default lowering path on GPU.

  Show description for 8a10ea2
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  8a10ea2

  Copy full SHA for 8a10ea2

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

  Show description for c63d403
  Google-ML-Automation

  committed
  c63d403

  Copy full SHA for c63d403

Commits on Dec 24, 2025

• [Pallas/Mosaic GPU] Make Mosaic GPU the default lowering backend for Pallas on GPU.

  Show description for af36ae2
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  af36ae2

  Copy full SHA for af36ae2

• [pmap] Update migrate pmap docs with additional user errors and performance notes.

  Show description for 206a7d2
  

  danielsuo

  authored and

  Google-ML-Automation

  committed
  206a7d2

  Copy full SHA for 206a7d2

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

  Show description for 1174c79
  Google-ML-Automation

  committed
  1174c79

  Copy full SHA for 1174c79

Commits on Dec 23, 2025

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

  Show description for 82ae1b1
  Google-ML-Automation

  committed
  82ae1b1

  Copy full SHA for 82ae1b1

Commits on Dec 22, 2025

• Adjust tile_n assert to accommodate 2 cta in tcgen05 blockscale mma.

  Show description for 5173d18
  Google-ML-Automation

  committed
  5173d18

  Copy full SHA for 5173d18

• [Mosaic] Extend tpu.pack_elementwise to support non-32-bit integers.

  Show description for 9fe0430
  

  WindQAQ

  authored and

  Google-ML-Automation

  committed
  9fe0430

  Copy full SHA for 9fe0430

• Add programmatic profiling test with session_id support in JAX.

  Show description for c25a514
  

  subhamsoni-google

  authored and

  Google-ML-Automation

  committed
  c25a514

  Copy full SHA for c25a514

• [Autotuner] Prepare the CuDNN fusion test for the new autotuner.

  Show description for 6d41fa0
  

  derdrdirk

  authored and

  Google-ML-Automation

  committed
  6d41fa0

  Copy full SHA for 6d41fa0

• [Mosaic GPU] Add `nvshmemx_cumodule_finalize` support.

  Show description for 2da9713
  

  allanrenucci

  authored and

  Google-ML-Automation

  committed
  2da9713

  Copy full SHA for 2da9713

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

  Show description for f29645b
  Google-ML-Automation

  committed
  f29645b

  Copy full SHA for f29645b

Commits on Dec 21, 2025

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

  Show description for 0ad46d9
  Google-ML-Automation

  committed
  0ad46d9

  Copy full SHA for 0ad46d9

Commits on Dec 20, 2025

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

  Show description for 6469f17
  Google-ML-Automation

  committed
  6469f17

  Copy full SHA for 6469f17

• Fix a broken sharded -> unreduced test

  Show description for 20bdca7
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  20bdca7

  Copy full SHA for 20bdca7

• Add replicated -> unreduced test coverage

  Show description for d3ed2f4
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  d3ed2f4

  Copy full SHA for d3ed2f4

• Support weakref_lru_cache.evict.

  Show description for dd7ff37
  

  pschuh

  authored and

  Google-ML-Automation

  committed
  dd7ff37

  Copy full SHA for dd7ff37

Commits on Dec 19, 2025

• [dep] make jax_default_dtype_bits config a no-op for JAX v0.9.0.

  Show description for 514b7c4
  

  Jake VanderPlas

  authored and

  Google-ML-Automation

  committed
  514b7c4

  Copy full SHA for 514b7c4

• Integrate LLVM at llvm/llvm-project@7d381f2a5634

  Show description for 4f0f93e
  Google-ML-Automation

  committed
  4f0f93e

  Copy full SHA for 4f0f93e

• Merge pull request #34065 from mattjj:custom-vjp3-revival

  Show description for 69518c5
  Google-ML-Automation

  committed
  69518c5

  Copy full SHA for 69518c5

• [hijax] upgrade VJPHiPrimitive.def_fwd to support symbolic zeros

  mattjj

  committed
  ec5015f

  Copy full SHA for ec5015f

• Merge pull request #34060 from jakevdp:interpolation-dep

  Show description for 3f8f860
  Google-ML-Automation

  committed
  3f8f860

  Copy full SHA for 3f8f860

• [dep] Remove deprecated jax.lib submodules for JAX v0.9.0.

  Show description for 82f02eb
  

  Jake VanderPlas

  authored and

  Google-ML-Automation

  committed
  82f02eb

  Copy full SHA for 82f02eb

• Remove pre-0.8.2 version guards.

  Show description for 1085d3a
  

  belitskiy

  authored and

  Google-ML-Automation

  committed
  1085d3a

  Copy full SHA for 1085d3a

• [dep] remove references to the already-deprecated interpolation argument

  jakevdp

  committed
  d6944d3

  Copy full SHA for d6944d3

• [dep] finalize deprecation of lax.dot positional arguments for JAX v0.9.0.

  Show description for f314722
  

  Jake VanderPlas

  authored and

  Google-ML-Automation

  committed
  f314722

  Copy full SHA for f314722

• [dep] remove deprecated `jax_safer_randint` configuration for JAX v0.9.0

  Show description for a9dac11
  

  Jake VanderPlas

  authored and

  Google-ML-Automation

  committed
  a9dac11

  Copy full SHA for a9dac11

• Merge pull request #34058 from jakevdp:finalize-deps

  Show description for f29de06
  Google-ML-Automation

  committed
  f29de06

  Copy full SHA for f29de06

• Enable using custom hermetic NCCL version.

  Show description for cfc997c
  

  ybaturina

  authored and

  Google-ML-Automation

  committed
  cfc997c

  Copy full SHA for cfc997c

• Merge pull request #33884 from melissawm:advanced-ad

  Show description for 0d0d491
  Google-ML-Automation

  committed
  0d0d491

  Copy full SHA for 0d0d491

• Merge pull request #33828 from rao-ashish:asrao/cross_host_transfer_batching

  Show description for a82d51d
  Google-ML-Automation

  committed
  a82d51d

  Copy full SHA for a82d51d

Pagination

Previous

Next