sig
  type typ = Standard of float | Nesterov of float | None
  val run :
    Owl_neural_optimise.Momentum.typ ->
    Owl_algodiff.S.t -> Owl_algodiff.S.t -> Owl_algodiff.S.t
  val default :
    Owl_neural_optimise.Momentum.typ -> Owl_neural_optimise.Momentum.typ
  val to_string : Owl_neural_optimise.Momentum.typ -> string
end