Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Feb 4, 2026

• Bump rules_ml_toolchain to enable customization of CUDA Core Compute Libraries repository

  Show description for a8832d1
  

  yuriivcs

  authored and

  Google-ML-Automation

  committed
  a8832d1

  Copy full SHA for a8832d1

• Move local_config_git to tensorflow

  Show description for 8eea21c
  

  ethanluoyc

  authored and

  Google-ML-Automation

  committed
  8eea21c

  Copy full SHA for 8eea21c

• [ReplicaGroupV3][Refactor][13/n] Remove optional V2 CollectiveDeviceList field on V1 CollectiveDeviceList.

  Show description for 9d67cd3
  

  Varcho

  authored and

  Google-ML-Automation

  committed
  9d67cd3

  Copy full SHA for 9d67cd3

• PR #37247: [xla:gpu] Clean up execution stream assignment

  Show description for 3bb4d57
  

  ezhulenev

  authored and

  Google-ML-Automation

  committed
  3bb4d57

  Copy full SHA for 3bb4d57

• Update ExponentBias to account for e8m0 which does not have denormals.

  Show description for 460a486
  

  18praveenb

  authored and

  Google-ML-Automation

  committed
  460a486

  Copy full SHA for 460a486

• Removed unused declaration from coordination service.

  Show description for d1f6568
  

  mwhittaker

  authored and

  Google-ML-Automation

  committed
  d1f6568

  Copy full SHA for d1f6568

• [XLA] Canonicalize module for lowering when doing separate compilation of instructions

  Show description for 9c4876b
  

  zvikinoza

  authored and

  Google-ML-Automation

  committed
  9c4876b

  Copy full SHA for 9c4876b

• Port everything in PjRtStreamExecutorLoadedExecutable over to use CommonPjRtLoadedExecutable::Execute().

  Show description for 0448d7b
  

  pschuh

  authored and

  Google-ML-Automation

  committed
  0448d7b

  Copy full SHA for 0448d7b

• PR #37252: [xla:gpu] Unify buffer_uses and buffers beteen Thunk and Command

  Show description for a00e63e
  

  ezhulenev

  authored and

  Google-ML-Automation

  committed
  a00e63e

  Copy full SHA for a00e63e

• Migrate buffer_donation_test to HloHardwareIndependentTestBase

  Show description for 8effab0
  Google-ML-Automation

  committed
  8effab0

  Copy full SHA for 8effab0

• Causing issues with sharding in JAX

  Show description for 7ab805e
  

  belitskiy

  authored and

  Google-ML-Automation

  committed
  7ab805e

  Copy full SHA for 7ab805e

• [XLA] Copy frontend attributes when handling kCall in SPMD partitioner.

  Show description for 768a1ce
  

  seherellis

  authored and

  Google-ML-Automation

  committed
  768a1ce

  Copy full SHA for 768a1ce

• PR #37157: [ROCm] Enable the xla-opt binary build

  Show description for e4a4ef6
  

  nurmukhametov

  authored and

  Google-ML-Automation

  committed
  e4a4ef6

  Copy full SHA for e4a4ef6

• Update backends factory for autotuner.

  Show description for d0180f6
  Google-ML-Automation

  committed
  d0180f6

  Copy full SHA for d0180f6

• Refactor autotuner_main to use DebugOptions.

  Show description for bda6bd2
  Google-ML-Automation

  committed
  bda6bd2

  Copy full SHA for bda6bd2

• Use HloInstruction unique ID in CopyThunk AsyncEvents.

  Show description for 4a5e8d4
  

  khasanovaa

  authored and

  Google-ML-Automation

  committed
  4a5e8d4

  Copy full SHA for 4a5e8d4

• PR #37081: Fix/generalize AR+Slice matching

  Show description for 4b53fb9
  

  jfc4050

  authored and

  Google-ML-Automation

  committed
  4b53fb9

  Copy full SHA for 4b53fb9

• Automated Code Change

  Show description for 4f8470c
  Google-ML-Automation

  committed
  4f8470c

  Copy full SHA for 4f8470c

• [XTile] Propagate MLIR diagnostics as tsl::Status in XTile codegen.

  Show description for aff46ee
  

  WillFroom

  authored and

  Google-ML-Automation

  committed
  aff46ee

  Copy full SHA for aff46ee

• [XTile] Move emitter from xla::gpu to xla::xtile namespace.

  Show description for 7958e64
  

  WillFroom

  authored and

  Google-ML-Automation

  committed
  7958e64

  Copy full SHA for 7958e64

• PR #37101: [XLA:GPU] Add conv kind assignment pass

  Show description for 6e6bc59
  

  Cjkkkk

  authored and

  Google-ML-Automation

  committed
  6e6bc59

  Copy full SHA for 6e6bc59

• PR #36553: [ROCm] Enable backends/gpu/autotuner unit tests on ROCm

  Show description for d20563b
  

  Eetusjo

  authored and

  Google-ML-Automation

  committed
  d20563b

  Copy full SHA for d20563b

• [XLA] Internal Migration of IndexingMap storage from MLIR::AffineMap to XLA::SymbolicMap

  Show description for 6677be5
  Google-ML-Automation

  committed
  6677be5

  Copy full SHA for 6677be5

• PR #37212: [ROCm] Clean up tag filters for rocm builds

  Show description for 7828280
  

  alekstheod

  authored and

  Google-ML-Automation

  committed
  7828280

  Copy full SHA for 7828280

• PR #36981: [xla:gpu] Fix comm split deadlocks

  Show description for a41e22a
  

  ezhulenev

  authored and

  Google-ML-Automation

  committed
  a41e22a

  Copy full SHA for a41e22a

• [XLA]Guard HoistTransposeOfReshape with a flag and enable it on GPU.

  Show description for 0d581d0
  

  thcmbs

  authored and

  Google-ML-Automation

  committed
  0d581d0

  Copy full SHA for 0d581d0

• PR #36769: Modify .clang-format header sorting priorities

  Show description for a6a7282
  

  ezhulenev

  authored and

  Google-ML-Automation

  committed
  a6a7282

  Copy full SHA for a6a7282

• The changes include

  Show description for 9cbde90
  

  pragun22

  authored and

  Google-ML-Automation

  committed
  9cbde90

  Copy full SHA for 9cbde90

• Modify RemoveEmptyPlanes to keep planes with stats.

  Show description for c19c239
  

  bmass02

  authored and

  Google-ML-Automation

  committed
  c19c239

  Copy full SHA for c19c239

• [XLA:MSA] Fix scheduling corner cases in MSA -

  Show description for bcf1864
  

  subhankarshah

  authored and

  Google-ML-Automation

  committed
  bcf1864

  Copy full SHA for bcf1864

• Add JAX version metadata to xspace, creating a metadata registry API to add metadata

  Show description for 5106efb
  Google-ML-Automation

  committed
  5106efb

  Copy full SHA for 5106efb

• Clear sharding on cloned instructions in HloConstantSplitter.

  Show description for 4e63e27
  

  ZixuanJiang

  authored and

  Google-ML-Automation

  committed
  4e63e27

  Copy full SHA for 4e63e27

• [PJRT C] Add `PJRT_Shardings` extension

  Show description for c6f67e1
  

  hyeontaek

  authored and

  Google-ML-Automation

  committed
  c6f67e1

  Copy full SHA for c6f67e1

• Make Shape required for BufferUse.

  Show description for 8b20a5b
  

  ermilovmaxim

  authored and

  Google-ML-Automation

  committed
  8b20a5b

  Copy full SHA for 8b20a5b

Commits on Feb 3, 2026

• Add register and invoke callbacks to PjRtCApiClient

  Show description for f9ede68
  

  hhb

  authored and

  Google-ML-Automation

  committed
  f9ede68

  Copy full SHA for f9ede68

Pagination

Previous

Next