def tr_adam2
import bendtorch@1.3.0.0/main.bend as Bendtorchimport Base
3538def tr_adam2(-P: Type, p: P, mm: P, vv: P, g: P, p1: F32, r: Array<F32> & F32) -> Train<P>:
3539 match r:
3540 case Tuple{xs, p2}: Train{p, g, TwoMom{mm, vv}, p1, p2, 0.0}