Commits

Commit history

1. Commits on Jul 8, 2026

   • Apply scaling flag defaults in from_flags instead of flag definitions

     Show description for 34aed43
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     34aed43View commit details

     Copy full SHA for 34aed43

     Browse repository at this point

   • perf(moe): K-outermost (kbsm) combine to drop the T(2,128) re-layout

     Show description for bf27087
     

     Zixuan Jiang

     authored and

     wangkuiyi

     committed
     bf27087View commit details

     Copy full SHA for bf27087

     Browse repository at this point

   • decoder: add hidden_state_partition_spec, decoupled from logits_partition_spec

     Show description for c5fe97e
     

     Zixuan Jiang

     authored and

     wangkuiyi

     committed
     c5fe97eView commit details

     Copy full SHA for c5fe97e

     Browse repository at this point

   • Split sharding API: with_sharding_constraint stays strict, maybe_shard goes best-effort.

     Show description for 7d1560b
     

     ds-hwang

     authored and

     wangkuiyi

     committed
     7d1560bView commit details

     Copy full SHA for 7d1560b

     Browse repository at this point

   • Enable custom TPU topology for 7x cluster in -fm scheduler

     Show description for 38eae3e
     

     Liang (AIML-Compute) He

     authored and

     wangkuiyi

     committed
     38eae3eView commit details

     Copy full SHA for 38eae3e

     Browse repository at this point

   • Add on_running_tick hook on the LWS runner

     Show description for caf0e76
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     caf0e76View commit details

     Copy full SHA for caf0e76

     Browse repository at this point

   • Add async create_pools_to / delete_pools_above for elastic scaling

     Show description for bb4d031
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     bb4d031View commit details

     Copy full SHA for bb4d031

     Browse repository at this point

   • Add HPA and MetricsCollection runner-level configurables

     Show description for c2da3b2
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     c2da3b2View commit details

     Copy full SHA for c2da3b2

     Browse repository at this point

   • fix bastion tests for bazel sandbox

     Show description for 6322fee
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     6322feeView commit details

     Copy full SHA for 6322fee

     Browse repository at this point

   • Reject expansion slots when parent isn't admitted

     Show description for d8070bf
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     d8070bfView commit details

     Copy full SHA for d8070bf

     Browse repository at this point

   • Add expansion slot scheduling for elastic autoscaling

     Show description for c8630f3
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     c8630f3View commit details

     Copy full SHA for c8630f3

     Browse repository at this point

   • Add ReplicaManager hook for elastic replica state

     Show description for 608317d
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     608317dView commit details

     Copy full SHA for 608317d

     Browse repository at this point

   • Add elastic scaling flags to launch command

     Show description for 479f190
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     479f190View commit details

     Copy full SHA for 479f190

     Browse repository at this point

   • Add ScalingSpec types and min-resource aggregation helpers

     Show description for a50cfb8
     

     Alexandre James

     authored and

     wangkuiyi

     committed
     a50cfb8View commit details

     Copy full SHA for a50cfb8

     Browse repository at this point

   • Add parallel option to CloudBuildBundler

     Show description for c1c31b0
     

     samos123

     authored and

     wangkuiyi

     committed
     c1c31b0View commit details

     Copy full SHA for c1c31b0

     Browse repository at this point

   • Avoid materializing full-sequence logits during prefill

     Show description for d72d49b
     

     ds-hwang

     authored and

     wangkuiyi

     committed
     d72d49bView commit details

     Copy full SHA for d72d49b

     Browse repository at this point

   • feat(summary_writer): align average_step_time with per-writer write_every_n_steps

     Show description for 813b80a
     

     Ralf Kaestner

     authored and

     wangkuiyi

     committed
     813b80aView commit details

     Copy full SHA for 813b80a

     Browse repository at this point

   • Truncate W&B tags to max length

     Show description for 9f96ba2
     

     Thomas Holleis

     authored and

     wangkuiyi

     committed
     9f96ba2View commit details

     Copy full SHA for 9f96ba2

     Browse repository at this point

   • golden_config.bzl: make golden_config_test usable cross-repo

     Show description for 840ba00
     

     Yanxin Jia

     authored and

     wangkuiyi

     committed
     840ba00View commit details

     Copy full SHA for 840ba00

     Browse repository at this point

   • Bumps array_record version to 0.8.3

     Show description for 668f6ef
     

     RsEnts

     authored and

     wangkuiyi

     committed
     668f6efView commit details

     Copy full SHA for 668f6ef

     Browse repository at this point

   • Bump ml-goodput-measurement from 0.0.16 to 0.2.0

     Show description for 0b9957f
     

     Jun Tian

     authored and

     wangkuiyi

     committed
     0b9957fView commit details

     Copy full SHA for 0b9957f

     Browse repository at this point

   • Fix CompositeLossMetrics._aggregate to use WeightedSummary.__add__

     Show description for 2e6c331
     

     ds-hwang

     authored and

     wangkuiyi

     committed
     2e6c331View commit details

     Copy full SHA for 2e6c331

     Browse repository at this point

   • Add generic gating override/emit hooks

     Show description for 8678670
     

     Suhong Moon

     authored and

     wangkuiyi

     committed
     8678670View commit details

     Copy full SHA for 8678670

     Browse repository at this point

   • fix(grad_accum): preserve plain scalar summaries during minibatch merge

     Show description for 0c911d9
     

     Ralf Kaestner

     authored and

     wangkuiyi

     committed
     0c911d9View commit details

     Copy full SHA for 0c911d9

     Browse repository at this point

   • fix(summary_writer): batch D2H transfers in summary writers

     Show description for 52903f5
     

     Ralf Kaestner

     authored and

     wangkuiyi

     committed
     52903f5View commit details

     Copy full SHA for 52903f5

     Browse repository at this point

   • Logging clipped token positions in `GradientMonitorAndClipLayer`

     Show description for 30cac9a
     

     Firenze11

     authored and

     wangkuiyi

     committed
     30cac9aView commit details

     Copy full SHA for 30cac9a

     Browse repository at this point

   • Stop on undefined loss

     Show description for 9e1e3f8
     

     adrianc-a

     authored and

     wangkuiyi

     committed
     9e1e3f8View commit details

     Copy full SHA for 9e1e3f8

     Browse repository at this point

   • Fix bfloat16 mantissa truncation in SlidingWindowKVCache position scatter

     Show description for 5c47ee6
     

     ds-hwang

     authored and

     wangkuiyi

     committed
     5c47ee6View commit details

     Copy full SHA for 5c47ee6

     Browse repository at this point

   • Preserve `partition_spec` When Enabling Sliding KV Cache

     Show description for 6c23573
     

     ds-hwang

     authored and

     wangkuiyi

     committed
     6c23573View commit details

     Copy full SHA for 6c23573

     Browse repository at this point

   • Fix merge summaries for non Summary objects

     Show description for 62c012a
     

     babych-d

     authored and

     wangkuiyi

     committed
     62c012aView commit details

     Copy full SHA for 62c012a

     Browse repository at this point

   • Migrate Causal Convolution Layers to AxLearn

     Show description for 0d355cc
     

     ds-hwang

     authored and

     wangkuiyi

     committed
     0d355ccView commit details

     Copy full SHA for 0d355cc

     Browse repository at this point

   • fix(checkpoint): inject AWS_ENDPOINT_URL into tensorstore S3 specs

     Show description for 754d4b1
     

     Charlie Yan

     authored and

     wangkuiyi

     committed
     754d4b1View commit details

     Copy full SHA for 754d4b1

     Browse repository at this point

   • Match Vocabulary protocol to seqio.Vocabulary's return type

     Show description for 0183c1b
     

     ds-hwang

     authored and

     wangkuiyi

     committed
     0183c1bView commit details

     Copy full SHA for 0183c1b

     Browse repository at this point

   • Revert KV quantisation prototype.

     Show description for 39344a9
     

     hwasungmars

     authored and

     wangkuiyi

     committed
     39344a9View commit details

     Copy full SHA for 39344a9

     Browse repository at this point

   • Add `carry`/`scan_kwargs` to `BaseStackedTransformerLayer`

     Show description for a6b9575
     

     rohitc33

     authored and

     wangkuiyi

     committed
     a6b9575View commit details

     Copy full SHA for a6b9575

     Browse repository at this point