Code

Add Rematerialization to Keras#20743

Code

Draft

divyashreepathihalli wants to merge 14 commits intokeras-team:masterkeras-team/keras:masterfrom divyashreepathihalli:rematerializationdivyashreepathihalli/keras:rematerializationCopy head branch name to clipboard

+512-1

Commits

Commits on Jan 9, 2025

• add remat op

  divyashreepathihalli

  committed
  b2e53f6

  Copy full SHA for b2e53f6

• update test

  divyashreepathihalli

  committed
  5be2e6f

  Copy full SHA for 5be2e6f

• remove print statements

  divyashreepathihalli

  committed
  7581b4a

  Copy full SHA for 7581b4a

• remove memory testing

  divyashreepathihalli

  committed
  7277150

  Copy full SHA for 7277150

• run api_gen.sh

  divyashreepathihalli

  committed
  073f46b

  Copy full SHA for 073f46b

Commits on Jan 13, 2025

• update docstring

  divyashreepathihalli

  committed
  3be014c

  Copy full SHA for 3be014c

• add remat scope

  divyashreepathihalli

  committed
  fa1460f

  Copy full SHA for fa1460f

• code reformat

  divyashreepathihalli

  committed
  fef369a

  Copy full SHA for fef369a

• update scope to return all configs

  divyashreepathihalli

  committed
  373a0dc

  Copy full SHA for 373a0dc

• add remat wrapper to layer

  divyashreepathihalli

  committed
  f1eb799

  Copy full SHA for f1eb799

Commits on Jan 14, 2025

• add output size mode

  divyashreepathihalli

  committed
  d8fa54f

  Copy full SHA for d8fa54f

Commits on Jan 15, 2025

• Merge branch 'keras-team:master' into rematerialization

  divyashreepathihalli

  authored
  38691f9

  Copy full SHA for 38691f9

Commits on Jan 22, 2025

• Merge branch 'keras-team:master' into rematerialization

  divyashreepathihalli

  authored
  62a24cf

  Copy full SHA for 62a24cf

• add activation mode to remat

  divyashreepathihalli

  committed
  8c30ed7

  Copy full SHA for 8c30ed7