PR #2446 Add Enzyme train function

Add Enzyme train function #2446

darsnack merged 21 commits into FluxML:master from wsmoses:master

Enable remaining enzyme test

16a896f0

Merge branch 'FluxML:master' into master

89ecf4c7

wsmoses force pushed from 7c0e8e39 to 4cb52296 2 years ago

mcabbott added enzyme

Add Enzyme train function

546ff165

wsmoses force pushed from 4cb52296 to 546ff165 2 years ago

wsmoses force pushed from 5e19e3c2 to 966be4a6 2 years ago

wsmoses force pushed from 966be4a6 to 3ce9e41f 2 years ago

Mark check_sizes as inactive

1d2052cf

wsmoses force pushed from 3ce9e41f to 1d2052cf 2 years ago

mcabbott commented on 2024-05-14

Adapt to suggestions

77c35044

Update train.jl

97a490f3

Update train.jl

d8f1ad83

tmp

e17bef83

try fix

eb62e15c

wsmoses closed this 1 year ago

wsmoses reopened this 1 year ago

Update train.jl

13c69890

Update train.jl

c07bea97

Update train.jl

c01ab6fe

darsnack requested changes on 2024-06-03

Fixup

0cc61907

fix

b7654bdf

kwargs

f4e300c4

fixup

b329ef14

Rearrange dispatch for enzyme train

7b8309de

Fix method ambiguity and skip some test for enzyme

419b99a9

Updated news

9a0f4867

Fix version in news and bump project.toml

a5afde4f

Merge branch 'master' into master

2796aac3

darsnack approved these changes on 2024-06-24

darsnack merged 0be44016 into master 1 year ago

Reviewers

darsnack

mcabbott

Assignees

No one assigned

Labels

enzyme

Milestone

No milestone

Flux.jl Add Enzyme train function #2446 Merged

Add Enzyme train function #2446

Flux.jl
Add Enzyme train function
#2446

Merged