type t = Owl_algodiff.S.tmodule Init : sig ... endmodule Input : sig ... endmodule Activation : sig ... endmodule Linear : sig ... endmodule LinearNoBias : sig ... endmodule Recurrent : sig ... endmodule LSTM : sig ... endmodule GRU : sig ... endmodule Conv2D : sig ... endmodule Conv3D : sig ... endmodule FullyConnected : sig ... endmodule MaxPool2D : sig ... endmodule AvgPool2D : sig ... endmodule Lambda : sig ... endmodule Dropout : sig ... endmodule Reshape : sig ... endmodule Flatten : sig ... endtype layer = | Input of Input.layer| Linear of Linear.layer| LinearNoBias of LinearNoBias.layer| LSTM of LSTM.layer| GRU of GRU.layer| Recurrent of Recurrent.layer| Conv2D of Conv2D.layer| Conv3D of Conv3D.layer| FullyConnected of FullyConnected.layer| MaxPool2D of MaxPool2D.layer| AvgPool2D of AvgPool2D.layer| Dropout of Dropout.layer| Reshape of Reshape.layer| Flatten of Flatten.layer| Lambda of Lambda.layer| Activation of Activation.layermodule Feedforward : sig ... end