GPU

[mloubout]

GPU enabled.

All network are not function free and each layer file implements

• forward(args..., network; kwargs...)
• inverse(args..., network; kwargs...)
• backward(args..., network; kwargs...)

and for some:

• forward_Y(args..., network; kwargs...)
• inverse_Y(args..., network; kwargs...)
• backward_inv(args..., network; kwargs...)

These function are set as property for all layers so that can still use layer.forward(X)

All layers have an inverse as inverse(layer) that flips forward/backward. backward only supported for the few layers that already habe a backward_inv implemented