Lux.jl icon indicating copy to clipboard operation
Lux.jl copied to clipboard

Published 20 hours ago • verified publisher icon LuxDL

Automatic Differentiation Backends

Open avik-pal opened this issue 2 years ago • 0 comments

Currently all our tests use Zygote. We should update test_gradient_correctness_fdm to compare across all AD packages and FiniteDifferences.

  • [x] Zygote
  • [x] Tracker
  • [x] ReverseDiff
  • [x] ForwardDiff
  • [ ] Enzyme
  • [ ] Diffractor (once it supports Reverse Mode AD properly)

(These are in no particular order)

avik-pal avatar Feb 24 '23 02:02 avik-pal