Commits

Branch selector

 master

User selector

qlzh727

Datepicker

All time

Commit History

Commits on Feb 14, 2024

• Update the distribution lib to use jax.numpy ops instead of np. (#19178)

  qlzh727

  committedFeb 14, 2024
  f38b96c

Commits on Feb 11, 2024

• Minor cleanup for the unnecessary tf.summary.record_if(True) (no_op) (#19170)

  qlzh727

  committedFeb 11, 2024
  08c4b04

Commits on Nov 29, 2023

• Add keras.backend.device() API for device scope (#18837)

  qlzh727

  committedNov 29, 2023
  1ebe1d0

Commits on Nov 20, 2023

• Fix list_device logic (#18800)

  qlzh727

  committedNov 20, 2023
  ee2252e

Commits on Nov 7, 2023

• Update image_classification_with_vision_transformer.py (#18738)

  qlzh727

  committedNov 7, 2023
  127a11d

• Update image_classification_with_vision_transformer.py (#18740)

  qlzh727

  committedNov 7, 2023
  0ef8a0f

Commits on Nov 2, 2023

• Add data input distribution for model+data parallel. (#18721)

  qlzh727

  committedNov 2, 2023
  cf9b2f3

Commits on Nov 1, 2023

• Multi-worker dataset distribution for model+data parallel distribution. (#18717)

  qlzh727

  committedNov 1, 2023
  a4b95f0

Commits on Oct 26, 2023

• Fix the loss scale optimizer for tf.distribute. (#18691)

  qlzh727

  committedOct 26, 2023
  85dcc8b

Commits on Oct 24, 2023

• Add multi-worker support for JAX training. (#18654)

  qlzh727

  committedOct 24, 2023
  ee8b1ea

• Add unit test for sync batch norm under distribution strategy. (#18677)

  qlzh727

  committedOct 24, 2023
  555644e

• Fix numpy conversion for tf MirroredVariable. (#18675)

  qlzh727

  committedOct 24, 2023
  fd48a49

Commits on Oct 14, 2023

• Add dataset related function for multi worker/process distribution. (#18602)

  qlzh727

  committedOct 14, 2023
  2f82a4d

Commits on Oct 12, 2023

• Add initialization logic for multi-worker/process distribution. (#18596)

  qlzh727

  committedOct 12, 2023
  1553dee

Commits on Oct 10, 2023

• Add multi-process/worker support for JAX distribution. (#18551)

  qlzh727

  committedOct 10, 2023
  33ef6c5

• Split the jax distribution backend test into separate file. (#18587)

  qlzh727

  committedOct 10, 2023
  3d92f71

Commits on Oct 6, 2023

• Update layout_map to use re.search instead of re.match. (#18555)

  qlzh727

  committedOct 6, 2023
  f1ed36d

• Add sharding constraint for the output tensor in the model (#18536)

  qlzh727

  committedOct 6, 2023
  c57e454

Commits on Sep 27, 2023

• Update layout map to support tuple/list as the TensorLayout value when inserting. (#18514)

  qlzh727

  committedSep 27, 2023
  44f1be3

Commits on Sep 25, 2023

• Formatting the error string.

  

  qlzh727

  authored and

  fchollet

  committedSep 25, 2023
  ece4177

• Disable the stateless_apply for TF based optimizer

  

  qlzh727

  authored and

  fchollet

  committedSep 25, 2023
  6e536f6

• Fix TF optimizer with distribution strategy.

  

  qlzh727

  authored and

  fchollet

  committedSep 25, 2023
  f23d482

Commits on Sep 20, 2023

• Memory optimization for jax trainer. (#888)

  

  qlzh727

  authored and

  fchollet

  committedSep 20, 2023
  3390bee

Commits on Sep 19, 2023

• Fix JAX RNN backend issue. (#924)

  

  qlzh727

  authored and

  fchollet

  committedSep 19, 2023
  a400bd6

Commits on Sep 18, 2023

• Update jax trainer function to save memory buffer. (#897)

  

  qlzh727

  authored and

  fchollet

  committedSep 18, 2023
  1956e73

• Separate the metrics variables from non-trainable variables. (#910)

  

  qlzh727

  authored and

  fchollet

  committedSep 18, 2023
  e3d0b8b

Commits on Sep 11, 2023

• Add optimizer weights to the model summary. (#859)

  

  qlzh727

  authored and

  fchollet

  committedSep 11, 2023
  d279ece

Commits on Sep 7, 2023

• Update the eager_build logic for jax trainer. (#845)

  

  qlzh727

  authored and

  fchollet

  committedSep 7, 2023
  953b83f

Commits on Sep 3, 2023

• Fix the distribution logic for the JAX trainer. (#832)

  

  qlzh727

  authored and

  fchollet

  committedSep 3, 2023
  1eb2606

Commits on Aug 30, 2023

• Fix the Jax distributed variable load issue. (#817)

  

  qlzh727

  authored and

  fchollet

  committedAug 30, 2023
  584352b

Commits on Aug 29, 2023

• Add a dummy init for distribution, so it will be treat as a package. (#809)

  

  qlzh727

  authored and

  fchollet

  committedAug 29, 2023
  2305a79

Commits on Aug 27, 2023

• Add model parallel distribution. (#797)

  

  qlzh727

  authored and

  fchollet

  committedAug 27, 2023
  277b208

Commits on Aug 26, 2023

• Add LayoutMap which will be used for model parallelism (#793)

  

  qlzh727

  authored and

  fchollet

  committedAug 26, 2023
  75b83f6

Commits on Aug 25, 2023

• Reimplement the data parallel distribution. (#785)

  

  qlzh727

  authored and

  fchollet

  committedAug 25, 2023
  ecf5d08

Commits on Aug 23, 2023

• Add implementation for Distribution API (#776)

  

  qlzh727

  authored and

  fchollet

  committedAug 23, 2023
  5e1c58b

Pagination

PreviousNext