14 Pull requests merged by 1 person

• Internal

  #426 merged Aug 16, 2025

• Hermetic CUDA was moved from XLA to rules_ml_toolchain open sourced repository. The current Hermetic CUDA directory will be removed from XLA soon.

  #409 merged Aug 16, 2025

• [JAX SC] Enable tracking of IDs stopped due to static buffer size limits.

  #410 merged Aug 15, 2025

• Internal

  #424 merged Aug 15, 2025

• Code update

  #413 merged Aug 15, 2025

• Improve warning message for Coo Buffer Size. Specify how to update the stats.

  #421 merged Aug 15, 2025

• Fix logging in _populate_feature_spec.

  #420 merged Aug 15, 2025

• [JAX SC] [18/n] [Minibatching] Add multi-host test for input preprocessing minibatch count.

  #412 merged Aug 14, 2025

• [JAX SC] Fix sentinel node handling in input preprocessing.

  #414 merged Aug 13, 2025

• [JAX SC] [17/n] [Minibatching] Add test for single-host minibatching and fix existing bugs.

  #411 merged Aug 13, 2025

• [JAX SC] [16/n] Sync minibatching required/split across tables.

  #408 merged Aug 12, 2025

• [JAX SC] [Minibatching] [14/n] Populate and utilize num_minibatches.

  #407 merged Aug 11, 2025

• [JAX SC] [15/n] Conditionally branch when minibatching required.

  #404 merged Aug 11, 2025

• [JAX SC] Move SortAndGroup* function to separate header file to allow templating.

  #406 merged Aug 11, 2025

9 Pull requests opened by 1 person

• [JAX SC] [19/n] [Minibatching] Add `fake_all_reduce` utility for multi-host testing.

  #415 opened Aug 13, 2025

• [JAX SC] Update XLA commit and fix absl::MutexLock usage.

  #416 opened Aug 14, 2025

• [JAX SC] [21/n] [Minibatching] Integrate `MinibatchingSyncService` into input preprocessing.

  #417 opened Aug 14, 2025

• [JAX SC] [22/n] [Minibatching] Enable multi-host tests for minibatching.

  #418 opened Aug 14, 2025

• [JAX SC] [20/n] [Minibatching] Introduce MinibatchingSyncService for host-level sync.

  #419 opened Aug 14, 2025

• [JAX SC] [23/n] [Minibatching] Add PjRt-based all-reduce implementation.

  #422 opened Aug 14, 2025

• [JAX SC] Add `static_single_minibatch` option to input preprocessing.

  #423 opened Aug 14, 2025

• [JAX SC] Integrate dropped ID counts into base input stats.

  #425 opened Aug 15, 2025

• [JAX SC] Upgrade to C++20 and use `std::identity`.

  #427 opened Aug 16, 2025