Commits

Breadcrumbs

History for

ontest_704424560

User selector

All users

Datepicker

All time

Commit History

Commits on Jan 3, 2025

• In the Orbax Model (OBM) mode of Orbax Export, saves TF SavedModel to a sub-folder instead of the main OBM folder.

  Show description for 3fa494f
  

  wangpengmit

  authored and

  Orbax Authors

  committed
  3fa494f

  Copy full SHA for 3fa494f

Commits on Dec 24, 2024

• Change the Tensorflow export function from lambda function to normal function definition.

  Show description for c227788
  

  Orbax Authors

  committed
  c227788

  Copy full SHA for c227788

Commits on Dec 3, 2024

• Removes the Tensorflow Module base class from the JaxModule.

  Show description for 4c02ff6
  

  Orbax Authors

  committed
  4c02ff6

  Copy full SHA for 4c02ff6

Commits on Dec 2, 2024

• Removes the build function from the OrbaxModule and moves much of this functionality to the OrbaxExport to match the logic in TensorflowExport.

  Show description for ad1b199
  

  Orbax Authors

  committed
  ad1b199

  Copy full SHA for ad1b199

Commits on Nov 26, 2024

• No public description

  Show description for 7c67abf
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  7c67abf

  Copy full SHA for 7c67abf

• No public description

  Show description for 0ae565d
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  0ae565d

  Copy full SHA for 0ae565d

Commits on Nov 22, 2024

• ServingConfig.extra_trackable_resources is now supported in the Orbax Model path when support_tf_resources is True.

  Show description for 1cb2786
  

  wangpengmit

  authored and

  Orbax Authors

  committed
  1cb2786

  Copy full SHA for 1cb2786

Commits on Nov 20, 2024

• Adds a new support_tf_resources flag (defaults to False) to jax2obm_kwargs to enable a new way to save TF (concrete) functions in Orbax Model that allows those functions to use TF resources (…

  Show description for 71b1ba0
  

  wangpengmit

  authored and

  Orbax Authors

  committed
  71b1ba0

  Copy full SHA for 71b1ba0

Commits on Nov 15, 2024

• No public description

  Show description for 0f9e562
  

  Orbax Authors

  committed
  0f9e562

  Copy full SHA for 0f9e562

• Updated Version

  Show description for 577ebf2
  

  Orbax Authors

  committed
  577ebf2

  Copy full SHA for 577ebf2

• No public description

  Show description for db18acb
  

  Orbax Authors

  committed
  db18acb

  Copy full SHA for db18acb

Commits on Nov 13, 2024

• No public description

  Show description for 733f0d4
  

  Orbax Authors

  committed
  733f0d4

  Copy full SHA for 733f0d4

Commits on Oct 31, 2024

• Print the model_param shape instead of the model_params.

  Show description for 62ae7b4
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  62ae7b4

  Copy full SHA for 62ae7b4

Commits on Oct 28, 2024

• Wires up ExportManager to Orbax Model.

  Show description for 59532b0
  

  wangpengmit

  authored and

  Orbax Authors

  committed
  59532b0

  Copy full SHA for 59532b0

• Adds support for TF pre/post-processing to ObmModule and ObmExport.

  Show description for 8ae894e
  

  wangpengmit

  authored and

  Orbax Authors

  committed
  8ae894e

  Copy full SHA for 8ae894e

Commits on Oct 18, 2024

• Small change to fix test breakage

  Show description for 8cf8779
  

  Orbax Authors

  committed
  8cf8779

  Copy full SHA for 8cf8779

• Adds the ability to specify a checkpoint path to the Orbax export path.

  Show description for 753209e
  

  Orbax Authors

  committed
  753209e

  Copy full SHA for 753209e

Commits on Oct 17, 2024

• Begins to wire up the new Orbax Model export path to allow exporting using the new Orbax Model pathway.

  Show description for 06b9bd0
  

  Orbax Authors

  committed
  06b9bd0

  Copy full SHA for 06b9bd0

Commits on Sep 25, 2024

• Adds NotImplemented errors to obm_module and adds related tests to jax_module_test.

  Show description for c62f130
  

  Orbax Authors

  committed
  c62f130

  Copy full SHA for c62f130

• Cleans up the duplicated logic in jax_module after moving much of it to tensorflow_module in a previous checkin. Updates the tests for jax_module to focus on its smaller scope. Cleans up some of th…

  Show description for ff9f38f
  

  Orbax Authors

  committed
  ff9f38f

  Copy full SHA for ff9f38f

Commits on Sep 24, 2024

• Moves TF dependent code from jax_module in to tensorflow_module and from jax_module_test to tensorflow_module_test. This code is duplicated for now and will be cleaned up in a subsequent cl.

  Show description for fb7272d
  

  Orbax Authors

  committed
  fb7272d

  Copy full SHA for fb7272d

Commits on Sep 19, 2024

• Moves JaxModule.DEFAULT_METHOD_KEY into constants.py and updates references.

  Show description for fb0fc0b
  

  Orbax Authors

  committed
  fb0fc0b

  Copy full SHA for fb0fc0b

Commits on Sep 12, 2024

• Refactors some logic out of jax_module and continues the work of isolating the TensorFlow dependencies in TF specific subclasses.

  Show description for e7fa273
  

  Orbax Authors

  committed
  e7fa273

  Copy full SHA for e7fa273

Commits on Sep 11, 2024

• Adds an abstract class to define the data module for serialization and two subclasses that will implement the module logic. A subclass for the existing TensorFlow SavedModel export and a subclass f…

  Show description for 9b4a88d
  

  Orbax Authors

  committed
  9b4a88d

  Copy full SHA for 9b4a88d

Commits on Sep 10, 2024

• Adds two stubbed classes that will handle each of the export versions.

  Show description for 1018397
  

  Orbax Authors

  committed
  1018397

  Copy full SHA for 1018397

• Add comments to clarify the effect of is_primary in TensorSpecWithDefault

  Show description for bfadcb8
  

  Orbax Authors

  committed
  bfadcb8

  Copy full SHA for bfadcb8

Commits on Sep 5, 2024

• Add the config_class.py to set orbax_export global configs.

  Show description for 565b032
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  565b032

  Copy full SHA for 565b032

Commits on Aug 23, 2024

• Make the saved jax_exported map support jax.grad with new option vjp_order.

  Show description for 03934ef
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  03934ef

  Copy full SHA for 03934ef

Commits on Aug 17, 2024

• Enable pytype checking for orbax/export tests.

  Show description for 7cbea1d
  

  Orbax Authors

  committed
  7cbea1d

  Copy full SHA for 7cbea1d

Commits on Aug 14, 2024

• Internal.

  Show description for e272923
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  e272923

  Copy full SHA for e272923

Commits on Jul 24, 2024

• Remove an unnecessary negative test case in the test for polymorphic shapes with symbolic constraints

  Show description for 593b921
  

  Orbax Authors

  committed
  593b921

  Copy full SHA for 593b921

Commits on Jul 22, 2024

• Fixed typo: 'allow_multi_axis_sharding_conslidation' -> 'allow_multi_axis_sharding_consolidation'.

  Show description for 11f2151
  

  Orbax Authors

  committed
  11f2151

  Copy full SHA for 11f2151

Commits on Jul 17, 2024

• No public description

  Show description for 7d2cd82
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  7d2cd82

  Copy full SHA for 7d2cd82

Commits on Jul 16, 2024

• Require jax version >=0.4.30 as the jax.export API was introduced in this release.

  Show description for f62c930
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  f62c930

  Copy full SHA for f62c930

Commits on Jul 15, 2024

• Fork the _get_key_name util from ocp.

  Show description for bca0655
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  bca0655

  Copy full SHA for bca0655

Pagination

Previous

Next