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

• Fix

  wsmoses

  committed
  113c47b

• Add missing file

  wsmoses

  committed
  5055232

• Change to enzymecore ext

  wsmoses

  committed
  8bc816d

• attempt fix

  wsmoses

  committed
  330c2b1

• Add missing file

  wsmoses

  committed
  9799133

• Also add gather

  wsmoses

  committed
  b8f4493

• Additional functions, tests, and fixes

  wsmoses

  committed
  e6e98b4

• Add pooling

  wsmoses

  committed
  c32de74

• Add dropout

  wsmoses

  committed
  0ccb6c9

• Fix scatter bug

  wsmoses

  committed
  72a9963

• fix pool

  wsmoses

  committed
  6e64553

• More fixups

  wsmoses

  committed
  256a4fb

• Add batchnorm derivatives

  wsmoses

  committed
  597bcd7

• rebase

  wsmoses

  committed
  1933a9a