Revert #528 when upstream issue is fixed #533
Description
#528 added a less efficient workaround for a Zygote-specific issue: FluxML/Zygote.jl#1464 The PR can be reverted once the upstream issue is fixed.
{{ message }}
#528 added a less efficient workaround for a Zygote-specific issue: FluxML/Zygote.jl#1464 The PR can be reverted once the upstream issue is fixed.