William Moses

Missing invoke handling causes infinite loop

14

See CI on 1.10 cc @vchuravy @gbaraldi

### Motivation and description Arises when differentiating this model. ``` using Enzyme, Flux, Random, Test Enzyme.API.runtimeActivity!(true) loss(model, x) = sum(model(x)) function test_model(model, x, mi) println(model) l = loss(model, x) Flux.reset!(model)...

Add EnzymeRules for batchnorm

4

Follow up to https://github.com/FluxML/NNlib.jl/pull/536 I made this a separate PR, because for efficiency sake, it is better to have an inplace version of batchnorm, but that wasn't an exported API....

Add Enzyme Forward mode custom rule

31

Wasn't sure how to integrate the sample test file into your infrastructure.