sig
  type typ =
      Uniform of float * float
    | Gaussian of float * float
    | Standard
    | Tanh
    | Custom of (int -> int -> float)
  val run : Owl_neural_neuron.Init.typ -> int -> int -> Owl_algodiff.S.t
  val to_string : Owl_neural_neuron.Init.typ -> string
end