Enzyme-JAX
Enzyme-JAX copied to clipboard
batchnorm gradient is always zero
https://buildkite.com/julialang/lux-dot-jl/builds/5545#0194f760-d7e2-43f7-af3c-30d3b921a8f1/158-1060