(module Owl_neural_optimise.Gradient)