Code

feat: enzyme autodiff helpers#954

Code

Draft

avik-pal wants to merge 6 commits intomainLuxDL/Lux.jl:mainfrom ap/ho-enzymeLuxDL/Lux.jl:ap/ho-enzymeCopy head branch name to clipboard

+262-46

Commits

Commits on Jan 11, 2025

• feat: add forward mode batched enzyme jacobian

  avik-pal

  committed
  1c57aaa

  Copy full SHA for 1c57aaa

• feat: add reverse mode batched enzyme jacobian

  avik-pal

  committed
  5985015

  Copy full SHA for 5985015

• feat: add vjp and jvp for Enzyme

  avik-pal

  committed
  0c1770e

  Copy full SHA for 0c1770e

• fix: avoid closures in batched_jacobian

  avik-pal

  committed
  c15ccb2

  Copy full SHA for c15ccb2

• test: add batched jacobian tests for enzyme

  avik-pal

  committed
  d045b23

  Copy full SHA for d045b23

• feat: initial support for reactant

  avik-pal

  committed
  3ed672d

  Copy full SHA for 3ed672d