Flux.jl
Add more `Duplicated` methods for Enzyme.jl support
#2471

Merged

Add more `Duplicated` methods for Enzyme.jl support #2471

mcabbott merged 38 commits into FluxML:master from mcabbott:enzyme3

mcabbott added gradients

mcabbott added enzyme

mcabbott force pushed from ef9d7b42 to a024b169 1 year ago

mcabbott force pushed from 7e155f8d to ca51a77b 1 year ago

mcabbott force pushed from 60acf271 to 63105484 1 year ago

mcabbott marked this pull request as ready for review 1 year ago

mcabbott commented on 2024-11-06

mcabbott force pushed from f8439991 to ca5a20f2 1 year ago

mcabbott force pushed from ca5a20f2 to 0f224536 1 year ago

wsmoses commented on 2024-11-11

add more Duplicated methods

370195fa

update macro to zero, show

5e80ab2b

make informative errors if you use Duplicated without loading Enzyme

cdba1370

note on macro

81541361

fix some tests

c6756560

add an Enzyme docs page

5b9380a0

tweaks & tests

9f92a8b2

typos

bd4634f4

news, docs

2644f10d

let Flux own the function update! to avoid piracy

29b9d87d

Revert "let Flux own the function update! to avoid piracy"

8b3754f5

demand Optimisers PR

385602d9

fixup

9087bb00

force depwarns

f23fd0d6

allow aux in withgradient

faf6a242

disallow Active

b2c495e0

disallow trivial Duplicated

b75f800b

don't use ReverseWithPrimal in gradient

50c0c920

tweak

48dc7173

giant post-rebase fixup after everything was moved around... all earl…

037d6ef1

mcabbott force pushed from a9424ab4 to 037d6ef1 1 year ago

clean up more rebase mess

317816c5

fix docs

0bef0d61

try out Ref for withgradient

ecef1f04

don't own `_make_zero!`

349dc99b

mcabbott force pushed from 976be71d to 349dc99b 1 year ago

add explicit errors for 2nd order

21d05c0f

Merge branch 'master' into enzyme3

4d460a48

CarloLucibello commented on 2024-11-24

more rebase problems

126e7bd7

CarloLucibello added breaking

teach Flux.state about Duplicated

55d24eb2

another explicit error for Zygote mistake

7a545310

ahem

a1380a0d

mcabbott commented on 2024-11-27

don't use Enzyme's make_zero!, fix some bugs

5c1650fb

maybe this works?

016cfe63

see if CI likes these

16de6481

turns out train! does have tests

a76dd612

enzyme tests

3e2c760c

mcabbott force pushed from 21a2d361 to 610e869c 1 year ago

mcabbott force pushed from 610e869c to d8a8880e 1 year ago

fix tests?

4b233c89

mcabbott force pushed from d8a8880e to 4b233c89 1 year ago

mcabbott added this to the v0.15 milestone 1 year ago

Merge branch 'master' into enzyme3

9afbf0ac

mcabbott commented on 2024-12-03

minor comments

c9ac5c0d

mcabbott merged f5d25e51 into master 1 year ago

mcabbott deleted the enzyme3 branch 1 year ago

Reviewers

CarloLucibello

wsmoses

Assignees

No one assigned

Labels

gradients breaking enzyme

Milestone

v0.15

Flux.jl Add more `Duplicated` methods for Enzyme.jl support #2471 Merged

Add more `Duplicated` methods for Enzyme.jl support #2471

Flux.jl
Add more `Duplicated` methods for Enzyme.jl support
#2471

Merged