All pull requests

Pull requests

• Give tied few-shot accuracies equal regularization ranks

  #226·sylvesterkaczmarek opened Oct 5, 2026Contributor·

  ·
  1

• Keep balanced routing auxiliary-loss gradients finite

  #224·sylvesterkaczmarek opened Oct 3, 2026Contributor·

  ·
  1

• Fix non-deterministic ordering assumption in input_pipeline_test.

  #222·copybara-service[bot] opened Oct 1, 2026Bot·

  ·
  1

• Stop inverse-square-root cooldowns at zero

  #220·sylvesterkaczmarek opened Sep 22, 2026Contributor·

  ·

• Match weight decay parameters by key

  #219·sylvesterkaczmarek opened Sep 22, 2026Contributor·

  ·

• Add pyrefly suppressions

  #213·copybara-service[bot] opened Aug 5, 2026Bot·

  ·
  1

• Minor fix after jax changes.

  #203·copybara-service[bot] opened Jun 20, 2025Bot·

  ·

• [JAX] Replace GSPMDSharding with NamedSharding.

  #202·copybara-service[bot] opened May 20, 2025Bot·

  ·

• [JAX] Replace GSPMDSharding with NamedSharding.

  #201·copybara-service[bot] opened Apr 24, 2025Bot·

  ·

• No public description

  #189·copybara-service[bot] opened Nov 7, 2024Bot·

  ·

• google-internal visibility change.

  #152·copybara-service[bot] opened Nov 13, 2023Bot·

  ·
  1

• google-internal visibility change

  #151·copybara-service[bot] opened Nov 13, 2023Bot·

  ·
  1

• Internal.

  #73·copybara-service[bot] opened Feb 7, 2023Bot·

  ·

• Internal changes.

  #60·copybara-service[bot] opened Dec 18, 2022Bot·

  ·

• Remove default for checkpoint argument.

  #43·copybara-service[bot] opened Sep 23, 2022Bot·

  ·

• Always pass checkpoint argument to TfDatasetIterator.

  #42·copybara-service[bot] opened Sep 23, 2022Bot·

  ·

• Internal changes.

  #30·copybara-service[bot] opened Aug 9, 2022Bot·

  ·

• internal

  #19·copybara-service[bot] opened Jul 11, 2022Bot·

  ·