7 Pull requests merged by 5 people

• Add RAttention

  #1302 merged Jul 17, 2025

• CompositeLossMetrics now performs a weighted sum of losses.

  #1251 merged Jul 17, 2025

• [Blackwell] Fix GPU Attention Test: test_sliding_window_mask

  #1297 merged Jul 16, 2025

• Fix the PartitionSpec for sequence sharding of mask attention bias

  #1301 merged Jul 15, 2025

• Retire remat_in_scan in repeat layer.

  #1298 merged Jul 15, 2025

• Handle malformed command gracefully

  #1296 merged Jul 15, 2025

• Use ubuntu 22.04 for dataflow vm

  #1294 merged Jul 13, 2025

8 Pull requests opened by 8 people

• [TEST, DO-NOT-MERGE] Trigger dependency problem between apache-beam and new tensorboard-plugin-profile.

  #1295 opened Jul 14, 2025

• Sliding window KV cache supports the flash attention path via standard attention fallback.

  #1299 opened Jul 15, 2025

• [NOT FOR MERGE] Integrate all new Goodput features in v12

  #1300 opened Jul 15, 2025

• Add page_pool argument

  #1303 opened Jul 16, 2025

• Allow override LIBTPU_INIT_ARGS from env

  #1306 opened Jul 17, 2025

• Do type casting before dispatching

  #1307 opened Jul 18, 2025

• Resolved failures during Conda environment creation in non-interactiv…

  #1308 opened Jul 18, 2025

• Add axlearn/experiments/logistic_regression_tutorial.py

  #1309 opened Jul 18, 2025

9 Unresolved conversations

Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.

• Workload Hang & Rolling Window Goodput Monitoring Support

  #1278 commented on Jul 14, 2025 • 6 new comments

• Adding LWS Integration

  #1174 commented on Jul 19, 2025 • 1 new comment

• [JAX API UPDATE] Update `utils.py`, as `_registry_with_keypaths` was removed

  #1161 commented on Jul 17, 2025 • 0 new comments

• Adding basic elastic training (pause-and-resume)

  #1256 commented on Jul 17, 2025 • 0 new comments

• Update NCCL configs for A3 Ultra and A4

  #1265 commented on Jul 14, 2025 • 0 new comments

• [Do not review] Upgrade Python to 3.12 and use Ubuntu 24.04

  #1280 commented on Jul 17, 2025 • 0 new comments

• Pathways Updates

  #1288 commented on Jul 17, 2025 • 0 new comments

• Upgrade Tensorboard Plugin Profile to enable Roofline and Framework Ops Stats views.

  #1292 commented on Jul 18, 2025 • 0 new comments

• [DRAFT] Scale Test Configurations

  #1293 commented on Jul 16, 2025 • 0 new comments