All pull requests

Pull requests

• Add CLI commands for local checks and copy collections in TrainEvaluator

  #1565·copybara-service[bot] opened Oct 12, 2026Bot·

  ·

• Internal

  #1556·copybara-service[bot] opened Oct 11, 2026Bot·

  ·

• Keep work unit identities when reloading a Kauldron sweep with XReload

  #1555·copybara-service[bot] opened Oct 10, 2026Bot·

  ·

• Add unit tests for Kauldron EMA optimizer state and checkpoint loading.

  #1554·copybara-service[bot] opened Oct 9, 2026Bot·

  ·

• Keep regularized cosine-loss gradients finite for zero embeddings

  #1550·sylvesterkaczmarek opened Oct 5, 2026·

  ·
  1

• Apply fractional masks to both standard-deviation moments

  #1544·sylvesterkaczmarek opened Oct 5, 2026·

  ·
  1

• Internal.

  #1536·copybara-service[bot] opened Sep 23, 2026Bot·

  ·

• Fix single-argument sum/min/max/prod in shape specs

  #1535·IncognitoQuack opened Sep 21, 2026·

  ·

• Eliminate throwaway eval step in CheckpointedEvaluator._get_init_aux_state.

  #1533·copybara-service[bot] opened Sep 18, 2026Bot·

  ·

• konfig: Add opt-in allowlist= to restrict what an untrusted config imports

  #1532·copybara-service[bot] opened Sep 15, 2026Bot·

  ·

• Preserve named-tuple paths and None semantics without JAX

  #1530·sylvesterkaczmarek opened Sep 10, 2026·

  ·

• Fix image summary loading in TensorBoard and MLDash.

  #1523·copybara-service[bot] opened Sep 1, 2026Bot·

  ·

• Support collection key prefixing in TrainEval dashboards and tracer sharding constraints.

  #1507·copybara-service[bot] opened Aug 15, 2026Bot·

  ·

• Code update

  #1503·copybara-service[bot] opened Jul 28, 2026Bot·

  ·

• Internal

  #1487·copybara-service[bot] opened Jul 20, 2026Bot·

  ·

• Some evaluation_mask fixes.

  #1481·copybara-service[bot] opened Jul 14, 2026Bot·

  ·

• Expose enable_async_checkpointing option in Kauldron Checkpointer.

  #1479·copybara-service[bot] opened Jul 11, 2026Bot·

  ·

• Enable pip caching in GitHub Actions workflow to prevent timeouts.

  #1464·copybara-service[bot] opened Jun 20, 2026Bot·

  ·
  1

• Flip default freeze to False in konfig.resolve and deprecate ImmutableDict.

  #1462·copybara-service[bot] opened Jun 20, 2026Bot·

  ·
  1

• fix missing hash sign for comment breaking kauldron/main.py

  #1455·chq-matteo opened Jun 14, 2026·

  ·
  1

• Add eval_until_step option to Kauldron evaluations

  #1422·copybara-service[bot] opened May 12, 2026Bot·

  ·

• fix: guard np.float128 for macOS ARM64 compatibility

  #1417·ac12644 opened Apr 5, 2026·3 tasks done·

  ·
  1

• Implement PEP 604 runtime object proxy | unions and static dummy generic indexing support for lazy kauldron configurations.

  #1412·copybara-service[bot] opened Mar 31, 2026Bot·

  ·
  1

• Make Kauldron modular when installing with pip.

  #1399·copybara-service[bot] opened Mar 19, 2026Bot·

  ·

• Extend KD cli commands.

  #1398·copybara-service[bot] opened Mar 18, 2026Bot·

  ·
  1