Commits

Branch selector

 master

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 25, 2023

• internal change

  Show description for aeab859
  

  ppwwyyxx

  authored and

  copybara-github

  committed
  aeab859

  Copy full SHA for aeab859

• Add three eval metrics to the v2 TPU Embedding API.

  Show description for 91eede8
  

  lingvo-bot

  authored and

  copybara-github

  committed
  91eede8

  Copy full SHA for 91eede8

• Consolidate logic and configuration between the TPUEmbeddingLayer v1 and v2 libraries.

  Show description for 81348d9
  

  lingvo-bot

  authored and

  copybara-github

  committed
  81348d9

  Copy full SHA for 81348d9

• Format tpu_embedding_layers.py with `black` so that future diffs are less confusing.

  Show description for 5ef0da8
  

  lingvo-bot

  authored and

  copybara-github

  committed
  5ef0da8

  Copy full SHA for 5ef0da8

Commits on Feb 24, 2023

• Some nits in advance of a CL unifying the params of the v1 and v2 TPU Embedding Layers.

  Show description for 1a0ff4b
  

  lingvo-bot

  authored and

  copybara-github

  committed
  1a0ff4b

  Copy full SHA for 1a0ff4b

• Adding functionality that makes it possible to save intermediate layer outputs in SequentialLayer.

  Show description for 982691a
  

  Eppens

  authored and

  copybara-github

  committed
  982691a

  Copy full SHA for 982691a

Commits on Feb 22, 2023

• Rename create_gda() into make_array() following GDA deprecation.

  Show description for 2eeaeba
  

  laurentes

  authored and

  copybara-github

  committed
  2eeaeba

  Copy full SHA for 2eeaeba

• Rename `in_axis_resources` and `out_axis_resources` with `in_shardings` and `out_shardings`. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't…

  Show description for 5576004
  

  lingvo-bot

  authored and

  copybara-github

  committed
  5576004

  Copy full SHA for 5576004

Commits on Feb 21, 2023

• Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

  Show description for 13d42cb
  

  lingvo-bot

  authored and

  copybara-github

  committed
  13d42cb

  Copy full SHA for 13d42cb

Commits on Feb 18, 2023

• Add BUILD dependency on jax:global_device_array to targets that use global device arrays.

  Show description for dc6359f
  

  lingvo-bot

  authored and

  copybara-github

  committed
  dc6359f

  Copy full SHA for dc6359f

Commits on Feb 17, 2023

• Use jax.Array API instead of the GlobalDeviceArray constructor since GDA is deprecated and going to be removed after march 15, 2023. jax.Array replacement for GDA should be 1:1 and should not chang…

  Show description for c5b9ab7
  

  lingvo-bot

  authored and

  copybara-github

  committed
  c5b9ab7

  Copy full SHA for c5b9ab7

• Report out tpu embedding learning rate as a summary tensor.

  Show description for 8acc7ff
  

  lingvo-bot

  authored and

  copybara-github

  committed
  8acc7ff

  Copy full SHA for 8acc7ff

• Silence some pytype errors.

  Show description for 694442e
  

  lingvo-bot

  authored and

  copybara-github

  committed
  694442e

  Copy full SHA for 694442e

Commits on Feb 14, 2023

• Add `self.variables` to _GetSelfVariablesDict and warn on variables that were tracked in `self.variables` but not `self._private_vars`.

  Show description for 7f69dcf
  

  lingvo-bot

  authored and

  copybara-github

  committed
  7f69dcf

  Copy full SHA for 7f69dcf

Commits on Feb 11, 2023

• Adhoc fix for qvars tracking in tf2

  Show description for 2c5ff9b
  

  lingvo-bot

  authored and

  copybara-github

  committed
  2c5ff9b

  Copy full SHA for 2c5ff9b

Commits on Feb 10, 2023

• Add sparse lookup capabilities to the TPU Embedding v2 layer.

  Show description for eaed632
  

  lingvo-bot

  authored and

  copybara-github

  committed
  eaed632

  Copy full SHA for eaed632

• This CL adds `tpu_embedding_layers_v2.py`, an updated and mostly config-compatible analogue to `tpu_embedding_layers_v2.py`, for use with the TPU Embedding V2 API. It may only be used in eager mode…

  Show description for 26ea608
  

  lingvo-bot

  authored and

  copybara-github

  committed
  26ea608

  Copy full SHA for 26ea608

Commits on Feb 9, 2023

• Log `BasicRecordYielder::Options`.

  Show description for 1ec8ae3
  

  lingvo-bot

  authored and

  copybara-github

  committed
  1ec8ae3

  Copy full SHA for 1ec8ae3

Commits on Feb 8, 2023

• internal changes

  Show description for d5ab366
  

  lingvo-bot

  authored and

  copybara-github

  committed
  d5ab366

  Copy full SHA for d5ab366

• Skip the cross_replica_sum for gradients when running with host driven

  Show description for f556c5d
  

  aaroey

  authored and

  copybara-github

  committed
  f556c5d

  Copy full SHA for f556c5d

• Augment HostDrivenTrainProgram with multihost infeed using tf.distribute.

  Show description for 064590f
  

  lingvo-bot

  authored and

  copybara-github

  committed
  064590f

  Copy full SHA for 064590f

Commits on Feb 7, 2023

• A couple nits in the learner and base_input_generator.

  Show description for eacd2dc
  

  lingvo-bot

  authored and

  copybara-github

  committed
  eacd2dc

  Copy full SHA for eacd2dc

• Short docstring nit

  Show description for e7d0a72
  

  lingvo-bot

  authored and

  copybara-github

  committed
  e7d0a72

  Copy full SHA for e7d0a72

• Docstring fix

  Show description for e00d115
  

  lingvo-bot

  authored and

  copybara-github

  committed
  e00d115

  Copy full SHA for e00d115

• Host driven executor: connect to tpu cluster before calling super().__init__,

  Show description for 70d333d
  

  aaroey

  authored and

  copybara-github

  committed
  70d333d

  Copy full SHA for 70d333d

Commits on Feb 6, 2023

• Recover the missing static shapes after GenericInput

  Show description for 7a68b8f
  

  lingvo-bot

  authored and

  copybara-github

  committed
  7a68b8f

  Copy full SHA for 7a68b8f

• Fix activation function unit test that did not seem to exercise the different activation functions.

  Show description for 6cc63fc
  

  lingvo-bot

  authored and

  copybara-github

  committed
  6cc63fc

  Copy full SHA for 6cc63fc

• Remove flaky test.

  Show description for c70687c
  

  lingvo-bot

  authored and

  copybara-github

  committed
  c70687c

  Copy full SHA for c70687c

Commits on Feb 3, 2023

• Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.

  Show description for 26809d5
  

  lingvo-bot

  authored and

  copybara-github

  committed
  26809d5

  Copy full SHA for 26809d5

• Nit: Follow-on to nested_map changelist, added one use of Keys() and one docstring improvement.

  Show description for ceb6e56
  

  lingvo-bot

  authored and

  copybara-github

  committed
  ceb6e56

  Copy full SHA for ceb6e56

• Access the variables in the host driven program train step function directly, and make them implicit inputs.

  Show description for 48dc6f9
  

  aaroey

  authored and

  copybara-github

  committed
  48dc6f9

  Copy full SHA for 48dc6f9

Commits on Feb 2, 2023

• Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.

  Show description for 3d3c736
  

  lingvo-bot

  authored and

  copybara-github

  committed
  3d3c736

  Copy full SHA for 3d3c736

• Nits: typos, type annotations, cleanups

  Show description for 2c95e03
  

  lingvo-bot

  authored and

  copybara-github

  committed
  2c95e03

  Copy full SHA for 2c95e03

• Add some useful/pythonic functions to NestedMap and moved tests into their own file.

  Show description for a167059
  

  lingvo-bot

  authored and

  copybara-github

  committed
  a167059

  Copy full SHA for a167059

• Internal Change

  Show description for af08764
  

  lingvo-bot

  authored and

  copybara-github

  committed
  af08764

  Copy full SHA for af08764

Pagination

Previous

Next