All pull requests

Pull requests

• [MTC] Add backup=True to save_checkpoint

  #6254·copybara-service[bot] opened Nov 11, 2026Bot·

  ·

• Install CI test dependencies through the Airlock PyPI mirror when reachable.

  #6253·copybara-service[bot] opened Nov 11, 2026Bot·

  ·
  1

  

• m3: add design doc for agents and contributors

  skip-ci-tests
  #6252·bvandermoon opened Nov 11, 2026Collaborator·4 tasks done·

  ·
  4

• Add DeepSeek-V3 671B FP8 (FSDP + EP) TPU7x recipe, 4k chips

  #6251·copybara-service[bot] opened Nov 11, 2026Bot·3 tasks done·

  ·
  1

• Add m3 rematerialization, hybrid loop barriers, and activation savepoints

  skip-ci-tests
  #6249·AusarYao28 opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  8

• Enforce highest matmul precision and fp32 input routing for fp32 gate and logits flags

  #6248·entrpn opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  11

• Training engine: AOT weight-sync staging estimate, standalone script cleanups, and tests

  #6246·copybara-service[bot] opened Nov 10, 2026Bot·4 tasks done·

  ·
  1

• feat(kimi-k3): add Kimi-K3 model (KDA/MLA hybrid, latent MoE, AttnRes, scan_layers)

  skip-ci-tests
  #6243·snehalv2002 opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  4

• gpu_collective_overlap_pass: handle fingerprint-prefixed PGLE profile keys

  skip-ci-tests
  #6242·xd-nv opened Nov 10, 2026·

  ·
  1

• Re-pin Gemma4 RL scripts to cloud-pathways 20261110-jax_0.10.1 digests.

  #6240·copybara-service[bot] opened Nov 10, 2026Bot·

  ·
  1

• Re-pin and bump JAX versions of Pathways image references to cloud-pathways.

  #6239·copybara-service[bot] opened Nov 10, 2026Bot·

  ·
  1

• Use cloud-pathways sidecar image repo in MaxText build script.

  #6238·copybara-service[bot] opened Nov 10, 2026Bot·

  ·
  1

• Update default Pathways images in configs, scripts, and wikis to cloud-pathways.

  #6237·copybara-service[bot] opened Nov 10, 2026Bot·

  ·
  1

• Make Profiler.deactivate() idempotent for nsys mode

  skip-ci-tests
  #6236·xd-nv opened Nov 10, 2026·4 tasks done·

  ·
  2

• Keep weight decay off frozen vision and audio encoders

  skip-ci-tests
  #6235·nuinashco-ks opened Nov 10, 2026·4 tasks·

  ·
  1

• test(vllm): check the direct MaxText-to-vLLM weight sync on real model trees

  #6234·chiajunglien opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  10

• Register colocated-python handlers before v0 checkpoint managers

  skip-ci-tests
  #6233·yrchen-tw opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  6

• Skip Orbax ArrayMetadata for colocated-python checkpoints to fix the elastic scale-up hang

  skip-ci-tests
  #6232·yrchen-tw opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  6

• Add standalone distributed RL trainer benchmark, JobSet renderer, and AOT compile helper

  #6229·copybara-service[bot] opened Nov 10, 2026Bot·

  ·
  1

  

• fix(goodput): flush GoodputRecorder after recording job_end_time

  skip-ci-tests
  #6228·KevinYenky opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  6

• fix(vllm): return the vLLM KV cache from more scanned decoder layers

  #6226·chiajunglien opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  4

• fix(vllm): interleave prefused MoE wi only for the TP fused kernel

  #6225·chiajunglien opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  6

• Add UntrainableParam for weights that are never trained

  skip-ci-tests
  #6224·nuinashco-ks opened Nov 10, 2026·4 tasks·

  ·
  2

• Stream Hugging Face checkpoints from GCS into MaxText (combined #5593-#5806)

  skip-ci-tests
  #6223·snehalv2002 opened Nov 10, 2026Collaborator·4 tasks done·

  ·
  3

• Introduce root AGENTS.md and the /maxtext-self-review local skill so coding

  #6222·copybara-service[bot] opened Nov 9, 2026Bot·

  ·