sig
  type typ = Full | Mini of int | Stochastic
  val run :
    Owl_neural_optimise.Batch.typ ->
    Owl_algodiff.S.t ->
    Owl_algodiff.S.t -> Owl_algodiff.S.t * Owl_algodiff.S.t
  val batches : Owl_neural_optimise.Batch.typ -> Owl_algodiff.S.t -> int
  val to_string : Owl_neural_optimise.Batch.typ -> string
end