Code

Add EnzymeRules for batchnorm#537

Code

Open

wsmoses wants to merge 15 commits intoFluxML:masterFluxML/NNlib.jl:masterfrom wsmoses:batchnormwsmoses/NNlib.jl:batchnormCopy head branch name to clipboard

+143

Commits

Commits on Oct 8, 2023

• Add EnzymeRule for conv

  wsmoses

  committed
  7f6af69

  Copy full SHA for 7f6af69

• Fix

  wsmoses

  committed
  113c47b

  Copy full SHA for 113c47b

• Add missing file

  wsmoses

  committed
  5055232

  Copy full SHA for 5055232

• Change to enzymecore ext

  wsmoses

  committed
  8bc816d

  Copy full SHA for 8bc816d

• attempt fix

  wsmoses

  committed
  330c2b1

  Copy full SHA for 330c2b1

• Add missing file

  wsmoses

  committed
  9799133

  Copy full SHA for 9799133

• Also add gather

  wsmoses

  committed
  b8f4493

  Copy full SHA for b8f4493

• Additional functions, tests, and fixes

  wsmoses

  committed
  e6e98b4

  Copy full SHA for e6e98b4

• Add pooling

  wsmoses

  committed
  c32de74

  Copy full SHA for c32de74

• Add dropout

  wsmoses

  committed
  0ccb6c9

  Copy full SHA for 0ccb6c9

• Fix scatter bug

  wsmoses

  committed
  72a9963

  Copy full SHA for 72a9963

• fix pool

  wsmoses

  committed
  6e64553

  Copy full SHA for 6e64553

• More fixups

  wsmoses

  committed
  256a4fb

  Copy full SHA for 256a4fb

• Add batchnorm derivatives

  wsmoses

  committed
  597bcd7

  Copy full SHA for 597bcd7

• rebase

  wsmoses

  committed
  1933a9a

  Copy full SHA for 1933a9a