dfenn

Are there plans to implement multi-GPU support? I'm considering re-implementing my PyTorch models using Knet, but multi-GPU support is an important factor since I'm working on a cluster. Thanks!

Thank you for the update. It sounds like it wouldn't be easily implementable at the moment (especially for a Julia beginner like myself), but could be something that's supported at...

Thanks for your helpful response. Working from your example, this seems to work ``` julia> Flux.gradient(x -> sum(Zygote.forwarddiff(z -> ForwardDiff.jacobian(model, z), x)), rand(5, 2))[1] 5×2 Matrix{Float64}: -0.00228453 -0.0116751 -0.00126774 -0.00494717...

Thanks for you response. I was able to get it working using `@views` for the convolutional layers. However, the same approach isn't working for dense layers, where the weights matrix...