Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 7, 2025

• Automated Code Change

  Show description for 88ce695
  

  Sax Authors

  authored and

  copybara-github

  committed
  88ce695

  Copy full SHA for 88ce695

• Automated Code Change

  Show description for 0659886
  

  Sax Authors

  authored and

  copybara-github

  committed
  0659886

  Copy full SHA for 0659886

Commits on Nov 5, 2025

• Internal.

  Show description for 0015ee2
  

  Sax Authors

  authored and

  copybara-github

  committed
  0015ee2

  Copy full SHA for 0015ee2

Commits on Oct 14, 2025

• Perform MMteb evaluation for Pax Dual Encoder models. To run the mmteb evaluation code, I let a few pax and jax packages and code be visiable to google3/learning/embedding/gembench/oss.

  Show description for 56752b2
  

  Sax Authors

  authored and

  copybara-github

  committed
  56752b2

  Copy full SHA for 56752b2

Commits on Oct 2, 2025

• Automated Code Change

  Show description for 64df664
  

  Sax Authors

  authored and

  copybara-github

  committed
  64df664

  Copy full SHA for 64df664

Commits on Sep 23, 2025

• Automated Code Change

  Show description for c36140b
  

  Sax Authors

  authored and

  copybara-github

  committed
  c36140b

  Copy full SHA for c36140b

Commits on Aug 11, 2025

• [bazel] build dependency update

  Show description for 035723d
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  035723d

  Copy full SHA for 035723d

• Add an accessor for _dummy_input_sample.

  Show description for ad3276b
  

  Sax Authors

  authored and

  copybara-github

  committed
  ad3276b

  Copy full SHA for ad3276b

Commits on Jul 31, 2025

• Migrate dependencies on jax experimental targets to `jax/experimental/BUILD`.

  Show description for e1c92c2
  

  danielsuo

  authored and

  copybara-github

  committed
  e1c92c2

  Copy full SHA for e1c92c2

Commits on Jul 18, 2025

• Automated Code Change

  Show description for 73c91c1
  

  Sax Authors

  authored and

  copybara-github

  committed
  73c91c1

  Copy full SHA for 73c91c1

Commits on Jul 16, 2025

• Add linking_mode option for sax server build rules.

  Show description for f9f7dd4
  

  Sax Authors

  authored and

  copybara-github

  committed
  f9f7dd4

  Copy full SHA for f9f7dd4

Commits on Jul 13, 2025

• Automated Code Change

  Show description for 3bd4270
  

  Sax Authors

  authored and

  copybara-github

  committed
  3bd4270

  Copy full SHA for 3bd4270

Commits on Jul 9, 2025

• Internal change

  Show description for 2f505fc
  

  Sax Authors

  authored and

  copybara-github

  committed
  2f505fc

  Copy full SHA for 2f505fc

Commits on Jul 8, 2025

• Internal change

  Show description for 0b3f7e8
  

  Sax Authors

  authored and

  copybara-github

  committed
  0b3f7e8

  Copy full SHA for 0b3f7e8

Commits on Jun 4, 2025

• Change logging from info to error so it's easier to find.

  Show description for c097bfd
  

  Sax Authors

  authored and

  copybara-github

  committed
  c097bfd

  Copy full SHA for c097bfd

Commits on May 21, 2025

• internal change

  Show description for 83fee97
  

  bignamehyp

  authored and

  copybara-github

  committed
  83fee97

  Copy full SHA for 83fee97

Commits on May 20, 2025

• allow continuous batching for mm input.

  Show description for e1f0fcc
  

  bignamehyp

  authored and

  copybara-github

  committed
  e1f0fcc

  Copy full SHA for e1f0fcc

Commits on Apr 30, 2025

• Automated Code Change

  Show description for 76d7905
  

  h-joo

  authored and

  copybara-github

  committed
  76d7905

  Copy full SHA for 76d7905

Commits on Apr 28, 2025

• Update JAX nightly index usage

  Show description for 23b50ad
  

  MichaelHudgins

  authored and

  copybara-github

  committed
  23b50ad

  Copy full SHA for 23b50ad

Commits on Apr 17, 2025

• Delete unused code.

  Show description for 0b4d080
  

  bixia1

  authored and

  copybara-github

  committed
  0b4d080

  Copy full SHA for 0b4d080

• Supports logging expert choices in Sax continuous batching.

  Show description for 0df13d1
  

  Sax Authors

  authored and

  copybara-github

  committed
  0df13d1

  Copy full SHA for 0df13d1

• Internal changes

  Show description for 64c8e7a
  

  bignamehyp

  authored and

  copybara-github

  committed
  64c8e7a

  Copy full SHA for 64c8e7a

Commits on Apr 15, 2025

• Update usage of cuda12_pip to cuda12 and cuda12_local to cuda12-local when installing jax

  Show description for 0dcb0a5
  

  nitins17

  authored and

  copybara-github

  committed
  0dcb0a5

  Copy full SHA for 0dcb0a5

Commits on Apr 9, 2025

• Automated Code Change

  Show description for 5975593
  

  Sax Authors

  authored and

  copybara-github

  committed
  5975593

  Copy full SHA for 5975593

Commits on Apr 8, 2025

• Add API to directly return status of custom model call.

  Show description for 3b94ce5
  

  Sax Authors

  authored and

  copybara-github

  committed
  3b94ce5

  Copy full SHA for 3b94ce5

Commits on Apr 7, 2025

• Internal changes.

  Show description for b8fef4a
  

  bignamehyp

  authored and

  copybara-github

  committed
  b8fef4a

  Copy full SHA for b8fef4a

• Resolve module name conficts and update the instrucitons.

  Show description for 5b7c8eb
  

  bignamehyp

  authored and

  copybara-github

  committed
  5b7c8eb

  Copy full SHA for 5b7c8eb

Commits on Apr 4, 2025

• Update requirments.txt for TPU.

  Show description for 44ed154
  

  bignamehyp

  authored and

  copybara-github

  committed
  44ed154

  Copy full SHA for 44ed154

Commits on Apr 2, 2025

• update requirements. Also make go compatible with version prior to 1.21.

  Show description for 771a095
  

  bignamehyp

  authored and

  copybara-github

  committed
  771a095

  Copy full SHA for 771a095

Commits on Mar 30, 2025

• Add pytorch gemma to saxml.

  Show description for c13a410
  

  bignamehyp

  authored and

  copybara-github

  committed
  c13a410

  Copy full SHA for c13a410

Commits on Mar 24, 2025

• Add a update_checkpoint subcommand to saxutil.

  Show description for a8b787b
  

  Sax Authors

  authored and

  copybara-github

  committed
  a8b787b

  Copy full SHA for a8b787b

• Enable checkpoint updates.

  Show description for a66a364
  

  Sax Authors

  authored and

  copybara-github

  committed
  a66a364

  Copy full SHA for a66a364

Commits on Mar 23, 2025

• Move the old Gemma API to `gemma/deprecated/`

  Show description for 22b2b0b
  

  Conchylicultor

  authored and

  copybara-github

  committed
  22b2b0b

  Copy full SHA for 22b2b0b

Commits on Mar 19, 2025

• Changed Gemma Configs to allow full Remat (reducing peak TPU Memory by ~50%)

  Show description for 42f357e
  

  Sax Authors

  authored and

  copybara-github

  committed
  42f357e

  Copy full SHA for 42f357e

Commits on Feb 26, 2025

• Internal change: rename of library to create SAX models on TF Hub.

  Show description for e8c80dd
  

  Sax Authors

  authored and

  copybara-github

  committed
  e8c80dd

  Copy full SHA for e8c80dd

Pagination

Previous

Next