Code

New training API#132

Code

Merged

seanmor5 merged 21 commits intomainelixir-nx/axon:mainfrom sm-new-training-engineelixir-nx/axon:sm-new-training-engineCopy head branch name to clipboardOct 19, 2021

+1,897-883

Commits

Commits on Oct 12, 2021

• Start of new training API

  seanmor5

  committed
  42e639b

  Copy full SHA for 42e639b

Commits on Oct 13, 2021

• Refactor API

  seanmor5

  committed
  4b8e0a5

  Copy full SHA for 4b8e0a5

Commits on Oct 16, 2021

• Finalize first draft of new training API

  seanmor5

  committed
  d7ea4bc

  Copy full SHA for d7ea4bc

• Formatting

  seanmor5

  committed
  a5d0363

  Copy full SHA for a5d0363

Commits on Oct 17, 2021

• Fix jit bugs

  seanmor5

  committed
  fea9046

  Copy full SHA for fea9046

• Add loop output transform

  seanmor5

  committed
  e5e9642

  Copy full SHA for e5e9642

• Update GAN example

  seanmor5

  committed
  041335b

  Copy full SHA for 041335b

• Simple gan for debugging

  seanmor5

  committed
  4a2069c

  Copy full SHA for 4a2069c

Commits on Oct 18, 2021

• Encapsulate layer_param extraction (#133)

  

  josevalim

  authored and

  seanmor5

  committed
  ca8be47

  Copy full SHA for ca8be47

• Wrap compiler error to provide more context (#135)

  

  josevalim

  authored and

  seanmor5

  committed
  266e5e5

  Copy full SHA for 266e5e5

• Don't grad through max subtract in softmax

  seanmor5

  committed
  2a25cd6

  Copy full SHA for 2a25cd6

• Fix failing test

  seanmor5

  committed
  8769518

  Copy full SHA for 8769518

• Refactor and some tests

  seanmor5

  committed
  2c5cb1c

  Copy full SHA for 2c5cb1c

• Formatting

  seanmor5

  committed
  b4f9520

  Copy full SHA for b4f9520

• Update examples

  seanmor5

  committed
  f7a0c44

  Copy full SHA for f7a0c44

• More documentation updates

  seanmor5

  committed
  4973da5

  Copy full SHA for 4973da5

• Add step factories

  seanmor5

  committed
  5c38467

  Copy full SHA for 5c38467

• Update documentation

  seanmor5

  committed
  969bf52

  Copy full SHA for 969bf52

Commits on Oct 19, 2021

• Fix GAN example

  seanmor5

  committed
  0425c0c

  Copy full SHA for 0425c0c

• Update GAN example

  seanmor5

  committed
  8437436

  Copy full SHA for 8437436

• Update examples

  seanmor5

  committed
  dc4944c

  Copy full SHA for dc4944c